视频按帧转化为图片

"""
视频按帧转换为图片
"""

import cv2
from PIL import Image
import numpy as np
import os

# cap = cv2.VideoCapture("D:\\MV-CS050-10GC (K32048854)\\Video_20230620152236107.avi")  # 获取视频对象
cap = cv2.VideoCapture(r"C:\Users\MSI-NB\Desktop\0720\Video_20230720144336498.avi")  # 获取视频对象
isOpened = cap.isOpened  # 判断是否打开
savepath = r'C:\Users\MSI-NB\Desktop\0720\daba_20230720_bullet\daba_20230720_bullet_pic'

# 视频信息获取
fps = cap.get(cv2.CAP_PROP_FPS)
print(fps)
siglenumber=0
imageNum = 0
sum=0
timef = 10  # 隔timef帧保存一张图片

while (isOpened):

    sum+=1

    (frameState, frame) = cap.read()  # 记录每帧及获取状态

    if frameState == True and (sum % timef==0):

        siglenumber+=1

        # 格式转变，BGRtoRGB
        # frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        # 转变成Image
        frame = Image.fromarray(np.uint8(frame))
        frame = np.array(frame)

        # RGBtoBGR满足opencv显示格式
        # frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
        name_image=[os.path.join(savepath,i) for i in os.listdir(savepath)]
        L=len(name_image)
        imageNum = L
        fileName = 'C:\\Users\\MSI-NB\\Desktop\\0720\\daba_20230720_bullet\\daba_20230720_bullet_pic\\daba0720_img_' + str(imageNum) + '.jpg'  # 存储路径
        cv2.imwrite(fileName, frame, [cv2.IMWRITE_JPEG_QUALITY, 100])
        print(fileName + " successfully write in")  # 输出存储状态

    elif frameState == False:
        break

print('finish!,此次生成', siglenumber, '张图片')

cap.release()