yapf

tensorlayer · Feb 7, 2021 · 2d96739 · 2d96739
1 parent 022e8c2
commit 2d96739
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tensorlayer/app/human_pose_estimation/LCN.py b/tensorlayer/app/human_pose_estimation/LCN.py
@@ -221,10 +221,10 @@ def cgcnn_train():
 
     # === Last layer ===
     input_size4 = int(output.get_shape()[1])
-    output = Mask_layer(in_channels=input_size4, out_channels=OUT_JOINTS*3, name=["w4", "b4"])(output)
+    output = Mask_layer(in_channels=input_size4, out_channels=OUT_JOINTS * 3, name=["w4", "b4"])(output)
 
     # === End linear model ===
-    output = End_layer()([input_layer,output])
+    output = End_layer()([input_layer, output])
 
     network = Model(inputs=input_layer, outputs=output)
 
@@ -329,4 +329,4 @@ def CGCNN(pretrained=True):
         restore_params(network, model_path='model/model.npz')
     else:
         network = cgcnn_train()
-    return network
+    return network