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)
3.略解
利用parser.add_argument设置input ,output,skip_frame,在default设置参数,
用cv2.imwrite写入图片,用int(cv2.IMWRITE_JPEG_QUALITY), 100])保证图片质量,