demo code review update 2

Author: alex9311

demo/inference-config.yaml

@@ -11,7 +11,7 @@ WORKERS: 24
 PRINT_FREQ: 100
 
 DATASET:
-  COLOR_RGB: FALSE
+  COLOR_RGB: true
   DATASET: 'coco'
   DATA_FORMAT: jpg
   FLIP: true

demo/inference.py

@@ -261,7 +261,8 @@ def main():
         # pose estimation
         box = pred_boxes[0]  # assume there is only 1 person
         center, scale = box_to_center_scale(box, cfg.MODEL.IMAGE_SIZE[0], cfg.MODEL.IMAGE_SIZE[1])
-        pose_preds = get_pose_estimation_prediction(pose_model, image_bgr, center, scale)
+        image_pose = image.copy() if cfg.DATASET.COLOR_RGB else image_bgr.copy()
+        pose_preds = get_pose_estimation_prediction(pose_model, image_pose, center, scale)
 
         new_csv_row = []
         for _, mat in enumerate(pose_preds[0]):