python做视频抽帧图_python视频按帧截取图片工具
描述：将一个视频流按帧数截取大量的图片
用途：AI的数据集制作，得到大量的图片，之后将其打标签
更改的地方
1.default--间隔的帧数 2.input/output--输入视频的路径、存放截取图片的路径 （将路径放入后面的‘ 中即可）前面加r可表示绝对路径 eg:
args = parser.parse_args([--input,rF:data_videoIMG_4395.MOV,--output,rF:data_rgb_pic7video])
直接上代码
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)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。
时间： 2019-07-22
