summaryrefslogtreecommitdiff
path: root/cv/seg/img2video.py
blob: c824c9c84932e30ef4e5a3c51cb4d255fcf4f699 (plain)
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
import cv2
import os
from tqdm import tqdm

image_folder = './video_seg/v2j'
video_name = 'video.avi'

images = [img for img in os.listdir(image_folder) if img.endswith(".jpg")]
frame = cv2.imread(os.path.join(image_folder, images[0]))
height, width, layers = frame.shape

video = cv2.VideoWriter(video_name, 0, 10, (width,height))
images.sort(key=lambda s: int(s.split('.')[0].split('-')[1]))
for image in tqdm(images):
    video.write(cv2.imread(os.path.join(image_folder, image)))

cv2.destroyAllWindows()
video.release()


# import os
# import moviepy.video.io.ImageSequenceClip
# image_folder = './video_seg/v2j'
# fps = 10
#
# image_files = [os.path.join(image_folder,img)
#                for img in os.listdir(image_folder)
#                if img.endswith(".jpg")]
# clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(image_files, fps=fps)
# clip.write_videofile('my_video.mp4')