1. 需求分析
参考:十行Python代码制作一个视频倒放神器,由于最近在学习 OpenCV ,因此试着使用 OpenCV 进行实现,学以致用(胡乱折腾)。
- 需要视频倒放,因此需要读取视频cv.VideoCapture;
- 获取视频每一帧的图片采用cv.VideoCapture.read();
- 使用列表保存每一帧的图片;
- 使用 cv.VideoWriter 循环倒序向视频中写入图片。
2. 实现效果
2.1 正常视频
2.2 倒放视频
3. 读取视频
- 使用 cv.VideoCapture 读取视频;
- 使用 cap.get 获取视频的帧率;
- 创建保存每一帧图像的列表 images;
- 检查视频捕获是否初始化成功 cap.isOpened();
- 读取下一帧视频图像 ret, frame = cap.read();
- 判断有没有读取到图片 if ret is True;
- 将读取到的图片存入图像列表中;
- 完成循环,关闭视频文件或设备,释放对象;
- 返回视频的图像列表和视频的帧率。
# 读取视频,将视频按照帧导出图片
def get_video_images(video_path):
cap = cv.VideoCapture(video_path)
fps = cap.get(cv.CAP_PROP_FPS)
images = []
while cap.isOpened():
ret, frame = cap.read() # 读取下一帧视频图像
if ret is True:
images.append(frame)
key = cv.waitKey(round(fps))
if key == ord('q'):
break
else:
break
cap.release()
return {"images": images, "fps": fps}
4. 写入视频
- 通过图片获取视频的宽高;
- 设置视频的格式 cv.VideoWriter_fourcc;
- 创建视频的写入对象 cv.VideoWriter;
- 倒序循环视频图像列表;
- 写入下一帧视频 writer.write;
- 完成循环,关闭视频写入,释放对象。
# 转MP4
def create_mp4(filename, fps, images):
h,w,c = images[0].shape
fourcc = cv.VideoWriter_fourcc(*'mp4v')
writer = cv.VideoWriter(filename, fourcc, fps, (w,h))
for frame in images[::-1]:
key = cv.waitKey(round(fps))
if key == ord('q'):
break
writer.write(frame)
writer.release()
5. 完整代码
'''
version: 1.0.0
Author: Rattenking
Date: 2023-01-31 10:33:16
'''
import cv2 as cv
# 读取视频,将视频按照帧导出图片
def get_video_images(video_path):
cap = cv.VideoCapture(video_path)
fps = cap.get(cv.CAP_PROP_FPS)
images = []
while cap.isOpened():
ret, frame = cap.read() # 读取下一帧视频图像
if ret is True:
images.append(frame)
key = cv.waitKey(round(fps))
if key == ord('q'):
break
else:
break
cap.release()
return {"images": images, "fps": fps}
# 转MP4
def create_mp4(filename, fps, images):
h,w,c = images[0].shape
fourcc = cv.VideoWriter_fourcc(*'mp4v')
writer = cv.VideoWriter(filename, fourcc, fps, (w,h))
for frame in images[::-1]:
key = cv.waitKey(round(fps))
if key == ord('q'):
break
writer.write(frame)
writer.release()
if __name__ == "__main__":
imgs = get_video_images('./images/Megamind.avi')
create_mp4('./images/Megamind.mp4', images=imgs.get("images"), fps=imgs.get("fps"))
6. 总结
- OpenCV 实现视频倒放,原理就是读取视频,获取每一帧的图像,倒转帧的顺序,存入新的视频;
- 明白其中的实现,还可以实现视频中间部分倒放、视频部分图像放大抖动等特效。