KevinQHLin commited on
Commit
e9f28ca
1 Parent(s): 9681396

Update run_on_video/video_extractor.py

Browse files
Files changed (1) hide show
  1. run_on_video/video_extractor.py +1 -1
run_on_video/video_extractor.py CHANGED
@@ -69,7 +69,7 @@ def vid2clip(model, vid_path, output_file,
69
  dirname = os.path.dirname(output_file)
70
  # if not os.path.exists(dirname):
71
  # print(f"Output directory {dirname} does not exists, creating...")
72
- os.makedirs(dirname)
73
  np.savez(os.path.join(output_file, 'vid.npz'), features=vid_features)
74
  else:
75
  print(f'{input_file}, failed at ffprobe.\n')
 
69
  dirname = os.path.dirname(output_file)
70
  # if not os.path.exists(dirname):
71
  # print(f"Output directory {dirname} does not exists, creating...")
72
+ os.makedirs(output_file, exist_ok=True)
73
  np.savez(os.path.join(output_file, 'vid.npz'), features=vid_features)
74
  else:
75
  print(f'{input_file}, failed at ffprobe.\n')