From 678fab50280b647d95213a9695d07c49542696f2 Mon Sep 17 00:00:00 2001 From: zhang Date: Sat, 21 May 2022 14:23:49 +0800 Subject: 0521 --- cv/seg/img2video.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 cv/seg/img2video.py (limited to 'cv/seg/img2video.py') diff --git a/cv/seg/img2video.py b/cv/seg/img2video.py new file mode 100644 index 0000000..c824c9c --- /dev/null +++ b/cv/seg/img2video.py @@ -0,0 +1,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') \ No newline at end of file -- cgit v1.2.3