python 用OpenCV 将图片转视频

news2025/2/2 1:05:28

import os
import cv2
import numpy as np

cv2.VideoWriter（）参数  cv2.VideoWriter() 是 OpenCV 中用于创建视频文件的类。它的参数如下：
filename：保存视频的文件名。
fourcc：指定视频编解码器的 FourCC 代码，用于将视频压缩成指定格式，例如：“XVID”、“MJPG”, "mp4v"等。可以使用 cv2.VideoWriter_fourcc() 函数来获取 FourCC 代码。
fps：指定视频帧率，即每秒显示的帧数。
frameSize：指定视频帧的大小，即视频的分辨率，可以使用 (width, height) 形式的元组来指定。
isColor：指定是否为彩色视频。如果为 True，则为彩色视频；如果为 False，则为灰度视频。
其中，前三个参数是必需的，后两个参数是可选的。例如：
fps = 25
cv2.VideoWriter('output.avi', cv2.VideoWriter_fourcc(*'XVID'), fps, (640, 480))

def resize_img():
    target_size = (1000, 1000)  # 所有图片缩放设置一致尺寸，目标尺寸
    path = './images'
    path_new = './images_new'
    if not os.path.exists(path_new):
        os.makedirs(path_new)
    filelists = []
    imglist = []
    for i in os.listdir(path):
        file_path = os.path.join(path, i)
        print(file_path)
        img = cv2.imdecode(np.fromfile(file_path, dtype=np.uint8), cv2.IMREAD_COLOR)
        size = img.shape
        h, w = size[0], size[1]
        target_h, target_w = target_size[1], target_size[0]
        # 确定缩放的尺寸
        scale_h, scale_w = float(h / target_h), float(w / target_w)
        print(f'scale_h:{scale_h}, scale_w:{scale_w}')
        scale = max(scale_h, scale_w) # 选择最大的缩放比率
        new_w, new_h = int(w / scale), int(h / scale)
        # 缩放后其中一条边和目标尺寸一致
        resize_img = cv2.resize(img, (new_w, new_h))
        # 图像上、下、左、右边界分别需要扩充的像素数目
        top = int((target_h - new_h) / 2)
        bottom = target_h - new_h - top
        left = int((target_w - new_w) / 2)
        right = target_w - new_w - left
        print(f'top:{top} bottom:{bottom} left:{left} right:{right}') 
        cv2.imwrite(os.path.join(path_new, f'new_{i}'), resize_img) # 写入本地文件
        # 填充至 target_w * target_h
        pad_img = cv2.copyMakeBorder(resize_img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=[0, 0, 0])
        # cv2.imshow('img', pad_img)
        # cv2.waitKey(1000)
        filelists.append(os.path.join(path_new, f'new_{i}'))
        imglist.append(pad_img)
    return filelists, imglist


def cut_img(scale):
    path = './images'
    path_new = './images_new'
    if not os.path.exists(path_new):
        os.makedirs(path_new)
    filelists = []
    imglist = []
    for i in os.listdir(path):
        file_path = os.path.join(path, i)
        print(file_path)
        img = cv2.imdecode(np.fromfile(file_path, dtype=np.uint8), cv2.IMREAD_COLOR)
        size = img.shape
        h, w = size[0], size[1]
        rate1 = scale.split(':')
        w1 = int((w - w * int(rate1[0]) / int(rate1[1])) / 2)
        w2 = int(w - (w - w * int(rate1[0]) / int(rate1[1])) / 2)
        resize_img = img[0:h, w1:w2]
        cv2.imwrite(os.path.join(path_new, f'new_{i}'), resize_img) # 写入本地文件
        filelists.append(os.path.join(path_new, f'new_{i}'))
        imglist.append(resize_img)
    return filelists, imglist



def image_to_video():
    scale = '1:1'  # 裁剪比例,并保持高度不变
    # scale = '3:4'
    # scale = '9:16'
    filelists, imglist = cut_img(scale) # 裁剪
    # filelists, imglist = resize_img() # 缩放
    fourcc = cv2.VideoWriter.fourcc(*'mp4v')
    im = cv2.imread(filelists[0])
    print(im.shape)
    shape1 = (im.shape[1], im.shape[0])    #需要转为视频的图片的尺寸, 视频的分辨率
    print('shape1:', shape1)
    fps = 1
    writer = cv2.VideoWriter('./output.mp4', fourcc, fps , shape1)
    # for file_path in filelists:
    #     print(file_path)
    #     img = cv2.imread(file_path)
    #     writer.write(img)

    for i in imglist:
        writer.write(i)
    writer.release()


image_to_video()

我们无法打开output3.mp4。这可能是因为文件类型不受支
持、文件扩展名不正确或文件已损坏，0xC00D36C4