opencv视频提取照片，帧数间隔提取，不限格式avi/mp4，无损截取

本文链接：https://blog.csdn.net/weixin_61623830/article/details/139291181

1.需求

使用openCv提取avi里的视频（不限制avi格式，只要opencv支持的格式都行，具体格式自行百度），可以每隔几帧保存一张。

可在图像分析场景使用，从视频或图像中提取有意义的信息。 OpenCV库可用于对视频执行多项操作。

所需模块：

import cv2
import argparse
import os

基本使用oepncv操作

VideoCapture(File_path):

read() :

imwrite(filename, img[， params]):

2.代码实现

import cv2
import argparse
import os
def parse_args():
    """
    Parse input arguments   ！！中文路径
    """
    parser = argparse.ArgumentParser(description='Process pic')
    parser.add_argument('--input', help='video to process',dest='input',  default='G:/Germany/HDvideo/G2.mp4', type=str)#dest=,None//G11     G:/Germany/HDvideo/Gm5.mp4   15/16/18
    parser.add_argument('--output', help='pic to store', dest='output', default='G:/Germany/photo/G222Ptest', type=str)#dest=None
    #default为间隔多少帧截取一张图片
    parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=1, type=int) #间隔提取

    #parser.add_argument('--quality',dest='photo_quality',default=96,type=int)
    #input为输入视频的路径 ，output为输出存放图片的路径
    args = parser.parse_args()  #['--input','','--output','']
    return args

 
def process_video(i_video, o_video, num):
    cap = cv2.VideoCapture(i_video)
    num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT) #获取总帧数 
    print(num_frame)
    expand_name = '.jpg'
    if not cap.isOpened():
        print("Please check the path.")
        return

    cnt = 0
    count = 0
    while True:   #1:
        ret, frame = cap.read()
        cnt += 1
      
        if cnt % num == 0:
            count += 1
            cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame,[int(cv2.IMWRITE_JPEG_QUALITY), 100]) # 设置保存参数，保证高画质----  默认95

   
 
        if not ret:
            break

    cap.release() #释放视频流对象
 
if __name__ == '__main__':
    args = parse_args()
    if not os.path.exists(args.output):
        os.makedirs(args.output)
    print('Called with args:')
    print(args)
    
    process_video(args.input, args.output, args.skip_frame)