fix pose demo multiprocessing

oliviermoliner · Jan 21, 2020 · a350373 · a350373
1 parent ee4cd7e
commit a350373
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/configs/pose_estimation/pose_demo.yaml b/configs/pose_estimation/pose_demo.yaml
@@ -1,7 +1,7 @@
 processor_cfg:
   type: "processor.pose_demo.inference"
   gpus: 1
-  worker_per_gpu: 2
+  worker_per_gpu: 1
   video_file: resource/data_example/skateboarding.mp4
   save_dir: "work_dir/pose_demo"
 

diff --git a/mmskeleton/processor/pose_demo.py b/mmskeleton/processor/pose_demo.py
@@ -52,6 +52,7 @@ def worker(inputs, results, gpu, detection_cfg, estimation_cfg, render_image):
             res['render_image'] = render(image, res['joint_preds'],
                                          res['person_bbox'],
                                          detection_cfg.bbox_thre)
+
         results.put(res)