Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,043 Bytes
113884e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import cv2
import imageio
def get_total_frames(video_path):
cap = cv2.VideoCapture(video_path)
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
cap.release()
return total_frames
def extract_frames(input_path, output_path, target_fps, selected_frames):
total_frames = get_total_frames(input_path)
video_reader = imageio.get_reader(input_path)
fps = video_reader.get_meta_data()['fps']
target_total_frames = selected_frames
frame_interval = max(1, int(fps / target_fps))
selected_indices = [int(i * frame_interval) for i in range(target_total_frames)]
target_frames = [video_reader.get_data(i) for i in selected_indices]
with imageio.get_writer(output_path, fps=target_fps) as video_writer:
for frame in target_frames:
video_writer.append_data(frame)
if __name__ == "__main__":
input_video_path = ""
output_video_path = ""
target_fps = 8
selected_frames = 24
extract_frames(input_video_path, output_video_path, target_fps, selected_frames)
|