From e3e55b74c22a4bad4830fe5a9f20ce5c4be7e01a Mon Sep 17 00:00:00 2001
From: JianHe0628 <jl02958@surrey.ac.uk>
Date: Fri, 8 Nov 2024 16:45:40 +0000
Subject: [PATCH] add try exception
---
Extract_HaMeR_Jsons.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/Extract_HaMeR_Jsons.py b/Extract_HaMeR_Jsons.py
index cbaba55..db12011 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)
--
GitLab