webcam test

main.py

@@ -117,8 +117,12 @@ def infer(video, check, kpt_thr, webcam=True):
                 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)                                                                                                                                              
-                out_files.append(fullname)
+                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)
 
             else:
                 out_files.extend(out_file)