Adding Try Catch

predict_multi_video.py

@@ -340,20 +340,25 @@ if __name__ == "__main__":
     for path in input_video_path:
         path = os.path.join(_parser().video_path, path)
         print("Processing video: ", path)
-        main(_parser().checkpoint_path, 
-            path,
-            _parser().output_dir,
-            _parser().video_start_frame,
-            _parser().video_end_frame,
-            _parser().word_data_json,
-            _parser().num_classes,
-            _parser().num_in_frames,
-            _parser().confidence,
-            _parser().batch_size,
-            _parser().stride,
-            _parser().topk,
-            _parser().resize_res,
-            _parser().datasetname,
-            _parser().include_embds,
-            _parser().include_probs
-        )
+        try:
+            main(_parser().checkpoint_path, 
+                path,
+                _parser().output_dir,
+                _parser().video_start_frame,
+                _parser().video_end_frame,
+                _parser().word_data_json,
+                _parser().num_classes,
+                _parser().num_in_frames,
+                _parser().confidence,
+                _parser().batch_size,
+                _parser().stride,
+                _parser().topk,
+                _parser().resize_res,
+                _parser().datasetname,
+                _parser().include_embds,
+                _parser().include_probs
+            )
+        except Exception as e:
+            print("Error in processing video: ", path)
+            print(e)
+            continue