Opencv处理MOT数据集，将图片帧转换为视频数据

"""处理MOT数据集，将帧序列转换为视频"""
import os
import cv2

fps = 30  # 输出视频帧率
size = (1920, 1080)  # 输出帧大小


def process(path):
    files = os.listdir(path)
    for file in files:
        imgs_dir = os.path.join(path, file) + './img1'
        fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # opencv3.0
        video_writer = cv2.VideoWriter(
            os.path.join(path, file) + './' + str(file) + '.mp4', fourcc, fps, size)
        imgs = os.listdir(imgs_dir)
        imgs.sort(key=lambda x: int(x[:-4]))  # 避免排序错误
        for img in imgs:
            pic = cv2.imread(imgs_dir + '/' + img)
            pic = cv2.resize(pic, size)
            video_writer.write(pic)
        video_writer.release()
        print(f"视频{file}.mp4已生成！")

if __name__ == "__main__":
    pics_path = './test'
    process(pics_path)