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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
from setup import *
def extract_frames(path: str) -> None:
# Clear old frames
for root, dirs, files in os.walk(frames_path):
for file in files:
os.remove(os.path.join(root, file))
# Get input video
video = cv2.VideoCapture(path)
current_frame = 0
while True:
ret, image = video.read()
# Save each frame
if ret != 1:
break
cv2.imwrite(os.path.join(frames_path, f"frame{current_frame}.png"), image)
current_frame += 1
video.release()
cv2.destroyAllWindows()
def to_video(path: str, frames: list) -> None:
if len(frames) == 0:
raise Exception("No frames were found.")
video = cv2.VideoCapture(input_filename)
frame_rate = video.get(cv2.CAP_PROP_FPS)
video.release()
cv2.destroyAllWindows()
height, width, layers = cv2.imread(os.path.join(path, frames[0])).shape
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
writer = cv2.VideoWriter(output_filename_no_audio, fourcc, frame_rate, (width, height))
for frame in frames:
writer.write(cv2.imread(os.path.join(path, frame)))
cv2.destroyAllWindows()
writer.release()
# adding audio
clip = VideoFileClip(output_filename_no_audio)
audio_clip = AudioFileClip(input_filename)
clip.audio = audio_clip
clip.write_videofile(output_filename)
def get_frames(path):
frames = []
for root, dirs, files in os.walk(path):
for file in files:
frames.append(file)
frames.sort(key=get_number)
return frames
|
