forked from z-x-yang/Segment-and-Track-Anything
-
Notifications
You must be signed in to change notification settings - Fork 0
/
img2vid.py
26 lines (20 loc) · 753 Bytes
/
img2vid.py
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
import cv2
import os
# set the directory containing the images
img_dir = './assets/840_iSXIa0hE8Ek'
# set the output video file name and codec
out_file = './assets/840_iSXIa0hE8Ek.mp4'
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
# get the dimensions of the first image
img_path = os.path.join(img_dir, os.listdir(img_dir)[0])
img = cv2.imread(img_path)
height, width, channels = img.shape
# create the VideoWriter object
out = cv2.VideoWriter(out_file, fourcc, 10, (width, height))
# loop through the images and write them to the video
for img_name in sorted(os.listdir(img_dir)):
img_path = os.path.join(img_dir, img_name)
img = cv2.imread(img_path)
out.write(img)
# release the VideoWriter object and close the video file
out.release()