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)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。