no mmdet

Dockerfile

@@ -17,7 +17,7 @@ RUN pip3  install torchvision --extra-index-url https://download.pytorch.org/whl
 RUN pip3  install --no-cache-dir --upgrade -U openmim
 RUN mim install --no-cache-dir --upgrade mmengine
 RUN mim install "mmcv>=2.0.1"
-RUN mim install "mmdet>=3.1.0"
+# RUN mim install "mmdet>=3.1.0"
 
 RUN git clone https://github.com/open-mmlab/mmpose.git
 

main_noweb.py

@@ -144,7 +144,7 @@ def pose2d(video, kpt_threshold):
 
     result_generator = human(video, 
                             vis_out_dir = add_dir,
-                            return_vis=True,
+                            #return_vis=True,
                             radius = 5,
                             thickness=4,
                             rebase_keypoint_height=True,
@@ -157,6 +157,9 @@ def pose2d(video, kpt_threshold):
 
     out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
     kpoints = glob.glob(os.path.join(add_dir, "*.json"))
+    print(os.listdir(glob.glob(os.path.join(add_dir, "*.mp4"))))
+    print(glob.glob(os.path.join(add_dir, "*.json")))
+    print(os.listdir(add_dir))
 
     return "".join(out_file), "".join(kpoints)
 
@@ -197,10 +200,10 @@ with block:
                         # Insert slider with kpt_thr
                         file_kpthr = gr.Slider(0, 1, value=0.3, label='Keypoint threshold')
                         with gr.Row():
-                            submit_pose_file = gr.Button("Make 2d pose estimation", variant="primary",show_progress=False)
-                            submit_pose3d_file = gr.Button("Make 3d pose estimation", variant="primary",show_progress=False)
-                            submit_hand_file = gr.Button("Make 2d hand estimation", variant="primary",show_progress=False)
-                            submit_detect_file = gr.Button("Detect and track objects", variant="primary",show_progress=False)
+                            submit_pose_file = gr.Button("Make 2d pose estimation", variant="primary")
+                            submit_pose3d_file = gr.Button("Make 3d pose estimation", variant="primary")
+                            submit_hand_file = gr.Button("Make 2d hand estimation", variant="primary")
+                            submit_detect_file = gr.Button("Detect and track objects", variant="primary")
 
                 with gr.Row():
                     video_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)