webcam test

main.py

@@ -113,16 +113,17 @@ def infer(video, check, kpt_thr, webcam=True):
            
         else:
             if webcam==True:
+                print("WEBCAM")
                 add_dir = str(uuid.uuid4())
                 vidname = video.split("/")[-1]
-                vis_out_dir = "/".join(["/".join(video.split("/")[:-1]), add_dir])
-                out_file = poses(inferencer, video, vis_out_dir, kpt_thr)    
-                fullname = os.path.join(vis_out_dir, vidname)         
-                if i == "Estimate human 3d poses":     
-                    fullname = fullname[:-4]+"mp4" #Change to .mp4                                                                                                                       
-                    out_files.append(fullname)
-                else:
-                    out_files.append(fullname)
+                vis_out_dir_web = "/".join(["/".join(video.split("/")[:-1]), add_dir])
+                out_file = poses(inferencer, video, vis_out_dir_web, kpt_thr)    
+                fullname = os.path.join(vis_out_dir_web, vidname)         
+                #if i == "Estimate human 3d poses":     
+                #    fullname = fullname[:-4]+"mp4" #Change to .mp4                                                                                                                       
+                #    out_files.append(fullname)
+                #else:
+                out_files.append(fullname)
 
             else:
                 out_files.extend(out_file)
@@ -184,4 +185,6 @@ if __name__ == "__main__":
 
 
 # 00000.mp4
-# 000000.mp4
+# 000000.mp4
+
+