【OpenCV】获得视频的帧数、FPS以及按帧数将图片保存到本地

视频操作

获得视频的帧数

import os
import cv2

video_cap = cv2.VideoCapture('./input/test_1.mp4')

frame_count = 0
all_frames = []
while(True):
    ret, frame = video_cap.read()
    if ret is False:
        break
    all_frames.append(frame)
    frame_count = frame_count + 1

print(frame_count)
print(len(all_frames))

获得视频的FPS

import cv2

if __name__ == '__main__':

    video = cv2.VideoCapture("./input/test_1.mp4")

    (major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.')

    if int(major_ver) >= 3:
        fps = video.get(cv2.CAP_PROP_FPS)
        print("Frames per second using video.get(cv2.CAP_PROP_FPS) : {0}".format(fps))
    else:
        fps = video.get(cv2.cv.CV_CAP_PROP_FPS)
        print("Frames per second using video.get(cv2.cv.CV_CAP_PROP_FPS): {0}".format(fps))
video.release()

按帧数将图片保存到本地

import cv2
import os


def save_img():
    video_path = './input'
    videos = os.listdir(video_path)
    for video_name in videos:
        file_name = video_name.split('.')[0]
        folder_name = video_path + file_name
        os.makedirs(folder_name, exist_ok=True)
        vc = cv2.VideoCapture(video_path + '/' + video_name)
        c = 0
        rval = vc.isOpened()

        while rval:
            c = c + 1
            rval, frame = vc.read()
            pic_path = folder_name + '/'
            if rval:
                cv2.imwrite(pic_path + str(c) + '.png', frame)
                cv2.waitKey(1)
            else:
                break
        vc.release()
        print('save_success')
        print(folder_name)
save_img()