要保存视频,您可以使用OpenCV库中的VideoWriter函数。下面是一个保存视频的函数示例:
import cv2
def save_video(input_file, output_file, fps=30.0):
# 读取输入视频文件
cap = cv2.VideoCapture(input_file)
# 获取输入视频文件的宽度和高度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 创建VideoWriter对象,用于保存输出视频文件
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter(output_file, fourcc, fps, (width, height))
while cap.isOpened():
ret, frame = cap.read()
if ret:
# 保存每一帧到输出视频文件
out.write(frame)
cv2.imshow('frame', frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()
在上述示例中,函数save_video
接收输入视频文件的路径(input_file),输出视频文件的路径(output_file)和帧率(fps)作为参数。它首先打开输入视频文件并获取其宽度和高度。然后,它创建一个VideoWriter对象,该对象使用指定的输出文件、FourCC编码、帧率和图像尺寸来保存视频。接下来,它循环读取每一帧并将其保存到输出视频文件中,直到所有帧都被处理完毕。最后,它释放资源并关闭窗口。
您可以使用以下代码来调用save_video
函数:
input_file = 'input_video.mp4'
output_file = 'output_video.mp4'
fps = 30.0
save_video(input_file, output_file, fps)
请确保在运行此代码之前,您已经安装了OpenCV库,并且输入视频文件存在。