背景
数据集采集不单有图片格式,也有视频格式,有时我们需要根据采集来的视频生成图片,再进行标注。所以这里我们使用Python自动给视频逐帧截图。
注意:由于摄像机录制的视频大多是XMF格式,需要使用格式工厂等软件转成mp4格式。
代码
直接上代码,可以根据自己的需求修改,比如多个视频文件依次截图等。
import cv2
import argparse
import os
def parse_args():
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)
parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=3, type=int)
args = parser.parse_args(['--input', r'D:\FFOutput\Clip_mp4\Clip0044.mp4', '--output', r'D:\FFOutput\Clip_capture44'])
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("this video has " + str(num_frame) + " frame.")
# expand_name = '.jpg'
if not cap.isOpened():
print("Please check the path.")
cnt = 0
count = 0
while 1:
ret, frame = cap.read()
cnt += 1
if cnt % num == 0:
count += 1
filename = "44" + "%04d.jpg" % count
# cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)
cv2.imwrite(os.path.join(o_video, filename), 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)
print("process is finished now")