将若干张图像变成视频 mp4 的代码

最新推荐文章于 2024-07-02 16:24:16 发布

原创最新推荐文章于 2024-07-02 16:24:16 发布 · 365 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

代码库专栏收录该内容

8 篇文章

订阅专栏

该代码段读取指定目录下的预测(predict)和目标(gt)图像文件，按比例缩放它们，然后将它们转换为RGB格式并存储为mp4视频。预测和目标视频分别保存在输出目录中。

import cv2
import imageio
import os

# left out variables：predict_dir 、predict_affix 、predict_suffix、gt_dir 、gt_suffix、output_dir
#

predict_list = []
gt_list = []

predict_path_list =[]
gt_path_list =[]

scale_percent = 60  # percent of original size
width = int(1280 * scale_percent / 100)
height = int(720 * scale_percent / 100)
dim = (width, height)


for i in range(21):
    predict_file_name = predict_affix+str(i)+predict_suffix
    predict_path = predict_dir+predict_file_name

    gt_file_name = str(i)+gt_suffix
    gt_path = gt_dir+gt_file_name


    predict_data = cv2.imread(predict_path)
    gt_data = cv2.imread(gt_path)

    # resize image
    predict_data = cv2.resize(predict_data, dim, interpolation=cv2.INTER_AREA)
    gt_data = cv2.resize(gt_data, dim, interpolation=cv2.INTER_AREA)

    predict_data = cv2.cvtColor(predict_data, cv2.COLOR_BGR2RGB)
    gt_data = cv2.cvtColor(gt_data, cv2.COLOR_BGR2RGB)

    predict_list.append(predict_data)
    gt_list.append(gt_data)

    predict_path_list.append(predict_path)
    gt_path_list.append(gt_path)


imageio.mimwrite(os.path.join(output_dir, 'predict.mp4'), predict_list, fps=5, quality=9 )  
imageio.mimwrite(os.path.join(output_dir, 'gt.mp4'), gt_list, fps=5, quality=9 )