change num workers and visualization

Author: yonigozlan

experiments/infinity_coco/hrnet/w48_384x288_adam_lr1e-3.yaml

@@ -4,10 +4,10 @@ CUDNN:
   DETERMINISTIC: false
   ENABLED: true
 DATA_DIR: ""
-GPUS: (0,)
+GPUS: (0, 1)
 OUTPUT_DIR: "output_infinity_coco"
 LOG_DIR: "log_infinity_coco"
-WORKERS: 1
+WORKERS: 12
 PRINT_FREQ: 10
 
 DATASET:
@@ -91,7 +91,7 @@ MODEL:
 LOSS:
   USE_TARGET_WEIGHT: true
 TRAIN:
-  BATCH_SIZE_PER_GPU: 40
+  BATCH_SIZE_PER_GPU: 20
   SHUFFLE: true
   BEGIN_EPOCH: 0
   END_EPOCH: 200

lib/utils/vis.py

@@ -43,15 +43,11 @@ def save_batch_image_with_joints(
             for joint, joint_vis in zip(joints, joints_vis):
                 joint[0] = x * width + padding + joint[0]
                 joint[1] = y * height + padding + joint[1]
-                if joint_vis[0]:
-                    if i_joint < 17:
-                        cv2.circle(
-                            ndarr, (int(joint[0]), int(joint[1])), 2, [255, 0, 0], 2
-                        )
-                    else:
-                        cv2.circle(
-                            ndarr, (int(joint[0]), int(joint[1])), 2, [0, 0, 255], 2
-                        )
+                # if joint_vis[0]:
+                if i_joint < 17:
+                    cv2.circle(ndarr, (int(joint[0]), int(joint[1])), 2, [255, 0, 0], 2)
+                else:
+                    cv2.circle(ndarr, (int(joint[0]), int(joint[1])), 2, [0, 0, 255], 2)
                 i_joint += 1
             k = k + 1
     img = cv2.cvtColor(ndarr, cv2.COLOR_BGR2RGB)