OpenCV-Python实战(7) —— OpenCV 实现抖音视频倒放效果

1. 需求分析

参考：十行Python代码制作一个视频倒放神器，由于最近在学习 OpenCV ，因此试着使用 OpenCV 进行实现，学以致用（胡乱折腾）。

1. 需要视频倒放，因此需要读取视频cv.VideoCapture；
2. 获取视频每一帧的图片采用cv.VideoCapture.read()；
3. 使用列表保存每一帧的图片；
4. 使用 cv.VideoWriter 循环倒序向视频中写入图片。

2. 实现效果

2.1 正常视频

2.2 倒放视频

3. 读取视频

1. 使用 cv.VideoCapture 读取视频；
2. 使用 cap.get 获取视频的帧率；
3. 创建保存每一帧图像的列表 images；
4. 检查视频捕获是否初始化成功 cap.isOpened()；
5. 读取下一帧视频图像 ret, frame = cap.read()；
6. 判断有没有读取到图片 if ret is True；
7. 将读取到的图片存入图像列表中；
8. 完成循环，关闭视频文件或设备，释放对象；
9. 返回视频的图像列表和视频的帧率。

# 读取视频，将视频按照帧导出图片
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. 写入视频

1. 通过图片获取视频的宽高；
2. 设置视频的格式 cv.VideoWriter_fourcc；
3. 创建视频的写入对象 cv.VideoWriter；
4. 倒序循环视频图像列表；
5. 写入下一帧视频 writer.write；
6. 完成循环，关闭视频写入，释放对象。

# 转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. 总结

1. OpenCV 实现视频倒放，原理就是读取视频，获取每一帧的图像，倒转帧的顺序，存入新的视频；
2. 明白其中的实现，还可以实现视频中间部分倒放、视频部分图像放大抖动等特效。