diff --git a/Extract_HaMeR_Jsons.py b/Extract_HaMeR_Jsons.py index cbaba55f9d0b9cc087015a14dd454cd2ef589a9e..db120114c55f2e4c0580e6a105e25964271a0b00 100644 --- a/Extract_HaMeR_Jsons.py +++ b/Extract_HaMeR_Jsons.py @@ -322,7 +322,11 @@ if __name__ == '__main__': if Folder_List[0].endswith('.mp4'): for vid_file in Folder_List: args.vid = vid_file - main(args, model, renderer, device, cpm) + try: + main(args, model, renderer, device, cpm) + except Exception as e: + print(f"Error in processing {vid_file}: {e}") + elif Folder_List[0].endswith('.png') or Folder_List[0].endswith('.jpg'): args.img_folder = args.in_folder main(args, model, renderer, device, cpm)