CV | Python + OpenCV 提取AVI视频中关键帧图像（修改版）

简介

从 OpenCV+PILLOW 提取AVI视频中关键帧图像 找到了一段使用OpenCV把AVI视频切分成静态图像，提取视频中的关键帧的程序。

但是实际测试时，发现有一个问题。视频里面原本的程序是彩色的（比如说模型表现是镀金的），但是按照该程序运行后，出来的图像变色了（金色的表面变为蓝色了）。

原因为: opencv画图的颜色通道，和PIL 以及 Plt画图的颜色通道是不同的。用cv2读取的图像，要想不变色的用PIL 读取，必须转换颜色通道。在画图前添加一句：

img_plt = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)     # cv2转plt画图

本文贴出的程序更改了源程序中的图像显示方法，直接用cv2绘图。

程序

import cv2

def splitFrames(videoFileName):
    cap = cv2. VideoCapture(videoFileName) # 打开视频文件
    num = 1
    while True:
        # success 表示是否成功，data是当前帧的图像数据；.read读取一帧图像，移动到下一帧
        success, data = cap.read()
        if not success:
            break
            
        cv2.imwrite('./figures/' +str(num)+".jpg", data)
        
        num = num + 1
        print(num)
    cap.release()

splitFrames('C:/Users/Taozi/Desktop/2019.04.30/模拟实验.avi')