python视频按帧截取图片工具
更新时间:2019年07月23日 14:13:00 作者:剑峰随心
这篇文章主要为大家详细介绍了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'])
直接上代码
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=None, type=str) parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str) #default为间隔多少帧截取一张图片 parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=100, 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) expand_name = '.jpg' if not cap.isOpened(): print("Please check the path.") cnt = 0 count = 0 while 1: ret, frame = cap.read() cnt += 1 # how # many # frame # to # cut if cnt % num == 0: count += 1 cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame) if not ret: break 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)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
相关文章
Python编程functools模块创建修改的高阶函数解析
本篇文章主要为大家介绍functools模块中用于创建、修改函数的高阶函数,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2021-09-09Python3+PyInstall+Sciter解决报错缺少dll、html等文件问题
这篇文章主要介绍了Python3+PyInstall+Sciter解决报错缺少dll、html等文件问题,本文给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下2019-07-07
最新评论