fix demo scripts

open-mmlab · Jul 14, 2023 · 0eb310c · 0eb310c
1 parent 6a23e2c
commit 0eb310c
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 13 deletions.
diff --git a/demo/body3d_pose_lifter_demo.py b/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()
 

diff --git a/demo/bottomup_demo.py b/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()

diff --git a/demo/topdown_demo_with_mmdet.py b/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()

diff --git a/projects/just_dance/README.md b/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