[Fix] Fix demo scripts (#2542)

Author: Ben-Louis

demo/body3d_pose_lifter_demo.py

@@ -452,10 +452,11 @@ def main():
 
                 video_writer.write(mmcv.rgb2bgr(frame_vis))
 
-            # press ESC to exit
-            if cv2.waitKey(5) & 0xFF == 27:
-                break
-            time.sleep(args.show_interval)
+            if args.show:
+                # press ESC to exit
+                if cv2.waitKey(5) & 0xFF == 27:
+                    break
+                time.sleep(args.show_interval)
 
         video.release()
 

demo/bottomup_demo.py

@@ -196,11 +196,12 @@ def main():
 
                 video_writer.write(mmcv.rgb2bgr(frame_vis))
 
-            # press ESC to exit
-            if cv2.waitKey(5) & 0xFF == 27:
-                break
+            if args.show:
+                # press ESC to exit
+                if cv2.waitKey(5) & 0xFF == 27:
+                    break
 
-            time.sleep(args.show_interval)
+                time.sleep(args.show_interval)
 
         if video_writer:
             video_writer.release()

demo/topdown_demo_with_mmdet.py

@@ -261,11 +261,12 @@ def main():
 
                 video_writer.write(mmcv.rgb2bgr(frame_vis))
 
-            # press ESC to exit
-            if cv2.waitKey(5) & 0xFF == 27:
-                break
+            if args.show:
+                # press ESC to exit
+                if cv2.waitKey(5) & 0xFF == 27:
+                    break
 
-            time.sleep(args.show_interval)
+                time.sleep(args.show_interval)
 
         if video_writer:
             video_writer.release()

projects/just_dance/README.md

@@ -17,7 +17,7 @@ We provide a Jupyter Notebook [`just_dance_demo.ipynb`](./just_dance_demo.ipynb)
 Users can simply run the following command to generate the comparison video:
 
 ```shell
-python process_video ${TEACHER_VIDEO} ${STUDENT_VIDEO}
+python process_video.py ${TEACHER_VIDEO} ${STUDENT_VIDEO}
 ```
 
 ### Gradio