diff --git a/predict_multi_video.py b/predict_multi_video.py index 561ad64f40c8614cae2920f73b94370be7ce765e..9e0d2f83c1813c6cb1d01a2054b288bc1c3166aa 100644 --- a/predict_multi_video.py +++ b/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