python视频按帧截取图片工具
本文实例为大家分享了python视频按帧截取图片工具的具体代码，供大家参考，具体内容如下
描述：将一个视频流按帧数截取大量的图片
用途：AI的数据集制作，得到大量的图片，之后将其打标签
更改的地方
1.default--间隔的帧数 2.input/output--输入视频的路径、存放截取图片的路径（将路径放入后面的‘'中即可）前面加r可表示绝对路径eg:
args=parser.parse_args(['--input',r'F:\data_video\IMG_4395.MOV','--output',r'F:data_rgb_pic\7video'])
直接上代码
importcv2
importargparse
importos
defparse_args():
"""
Parseinputarguments
"""
parser=argparse.ArgumentParser(description='Processpic')
parser.add_argument('--input',help='videotoprocess',dest='input',default=None,type=str)
parser.add_argument('--output',help='pictostore',dest='output',default=None,type=str)
#default为间隔多少帧截取一张图片
parser.add_argument('--skip_frame',dest='skip_frame',help='skipnumberofvideo',default=100,type=int)
#input为输入视频的路径，output为输出存放图片的路径
args=parser.parse_args(['--input','','--output',''])
returnargs
defprocess_video(i_video,o_video,num):
cap=cv2.VideoCapture(i_video)
num_frame=cap.get(cv2.CAP_PROP_FRAME_COUNT)
expand_name='.jpg'
ifnotcap.isOpened():
print("Pleasecheckthepath.")
cnt=0
count=0
while1:
ret,frame=cap.read()
cnt+=1
#how
#many
#frame
#to
#cut
ifcnt%num==0:
count+=1
cv2.imwrite(os.path.join(o_video,str(count)+expand_name),frame)
ifnotret:
break
if__name__=='__main__':
args=parse_args()
ifnotos.path.exists(args.output):
os.makedirs(args.output)
print('Calledwithargs:')
print(args)
process_video(args.input,args.output,args.skip_frame)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。