import os
import cv2
import numpy as npcv2.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.shapeh, 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 - topleft = int((target_w - new_w) / 2)right = target_w - new_w - leftprint(f'top:{top} bottom:{bottom} left:{left} right:{right}') cv2.imwrite(os.path.join(path_new, f'new_{i}'), resize_img) # 写入本地文件# 填充至 target_w * target_hpad_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, imglistdef 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.shapeh, 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, imglistdef 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 = 1writer = 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
如果生成的视频报这个错,一般是视频的分辨率 设置的不匹配