帧序列合成视频

# -*- coding: utf-8 -*-
# @Time : 2022/6/23 11:02
# @Author : lc
# @File : syn_video.py
# @Software: PyCharm
# @Function: Compositing a sequence of frames into a video

import os
import cv2


def comVideo(frame_dir, video_save, fps):
    imglist = sorted(os.listdir(frame_dir))
    img = cv2.imread(os.path.join(frame_dir, imglist[0]))
    H, W, D = img.shape
    img_size = (W, H)
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    # fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
    videoWriter = cv2.VideoWriter(video_save, fourcc, fps, img_size)
    for image in imglist:
        img_name = os.path.join(frame_dir, image)
        frame = cv2.imread(img_name)
        videoWriter.write(frame)
        print(img_name.split('/')[-1])
    videoWriter.release()
    print('Finish!')


if __name__ == "__main__":
    fps = 30
    frame_dir = r"./frame/"
    video_save = r'./video/com.mp4'
    comVideo(frame_dir, video_save, fps)

# -*- coding: utf-8 -*-
# @Time : 2022/6/23 11:02
# @Author : lc
# @File : syn_video.py
# @Software: PyCharm
# @Function: Compositing a sequence of frames into a video

import os
import cv2


def comVideo(frame_dir, save_dir, fps=30):
    file_List = os.listdir(frame_dir)
    for file in file_List:
        img_dir = os.path.join(frame_dir, file)
        img_list = sorted(os.listdir(img_dir))
        img_path0 = os.path.join(img_dir, img_list[0])
        img0 = cv2.imread(img_path0)
        H, W, D = img0.shape
        img_size = (W, H)
        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
        save_path = save_dir + file + '.mp4'
        if os.path.exists(save_path):
            name = save_path.split('/')[-1]
            print("{} existed!".format(name))
            continue
        videoWriter = cv2.VideoWriter(save_path, fourcc, fps, img_size)
        for img in img_list:
            img_path = os.path.join(img_dir, img)
            frame = cv2.imread(img_path)
            videoWriter.write(frame)
        videoWriter.release()
        print("{}.mp4 saved!".format(file))


if __name__ == "__main__":
    fps = 30
    frame_dir = r"./pha_hd/"
    save_dir = r"./video/pha/"
    comVideo(frame_dir, save_dir, fps)