jhj0517 commited on
Commit
9d18995
1 Parent(s): f3cdbdc

Enable overwriting

Browse files
Files changed (1) hide show
  1. modules/video_utils.py +6 -2
modules/video_utils.py CHANGED
@@ -22,6 +22,7 @@ def extract_frames(
22
 
23
  command = [
24
  'ffmpeg',
 
25
  '-i', vid_input,
26
  '-q:v', str(quality),
27
  '-start_number', str(start_number),
@@ -31,12 +32,13 @@ def extract_frames(
31
  try:
32
  subprocess.run(command, check=True)
33
  except subprocess.CalledProcessError as e:
34
- logger.exception("Error occured while extracting frames from the video")
35
  raise f"An error occurred: {str(e)}"
36
 
37
 
38
  def get_frames_from_dir(vid_dir: str,
39
  available_extensions: Optional[Union[List, str]] = None) -> List:
 
40
  if available_extensions is None:
41
  available_extensions = [".jpg", ".jpeg", ".JPG", ".JPEG"]
42
 
@@ -50,7 +52,9 @@ def get_frames_from_dir(vid_dir: str,
50
  if not frame_names:
51
  return []
52
  frame_names.sort(key=lambda x: int(os.path.splitext(x)[0]))
53
- return frame_names
 
 
54
 
55
 
56
  def clean_image_files(image_dir: str):
 
22
 
23
  command = [
24
  'ffmpeg',
25
+ '-y', # Enable overwriting
26
  '-i', vid_input,
27
  '-q:v', str(quality),
28
  '-start_number', str(start_number),
 
32
  try:
33
  subprocess.run(command, check=True)
34
  except subprocess.CalledProcessError as e:
35
+ logger.exception("Error occurred while extracting frames from the video")
36
  raise f"An error occurred: {str(e)}"
37
 
38
 
39
  def get_frames_from_dir(vid_dir: str,
40
  available_extensions: Optional[Union[List, str]] = None) -> List:
41
+ """Get image file paths list from the dir"""
42
  if available_extensions is None:
43
  available_extensions = [".jpg", ".jpeg", ".JPG", ".JPEG"]
44
 
 
52
  if not frame_names:
53
  return []
54
  frame_names.sort(key=lambda x: int(os.path.splitext(x)[0]))
55
+
56
+ frames = [os.path.join(vid_dir, name) for name in frame_names]
57
+ return frames
58
 
59
 
60
  def clean_image_files(image_dir: str):