本文共 1002 字,大约阅读时间需要 3 分钟。
制作数据集时,需要将视频转存成图片,可以使用python或者matlab实现。
一 、python的转换小代码:
# -*- encoding: utf-8 -*-import cv2import osimages = 'G:/Images_PATH/'if not os.path.exists(images): os.mkdir(images)cap = cv2.VideoCapture("G:/Videos_PATH/1.mp4")c = 0# 可以按照指定帧间隔保存图片time = 10 while (1): success, frame = cap.read() if success: if(c%time ==0): cv2.imwrite(images + str(c) + '.jpg', frame) c = c + 1 print('image:',c) else: breakcap.release()
二 、matlab的转换小代码:
function getPictureFromMp4(video_file,pictureDir) video=VideoReader(video_file); frame_number=floor(video.Duration * video.FrameRate); for i=1:frame_number %每100帧存储一张图片 if rem(i,100)==0 image_name=strcat('11',num2str(i)); image_name=strcat(image_name,'.jpg'); I=read(video,i); imwrite(I,[pictureDir,'/',image_name],'jpg'); I=[]; end end end
使用使用matlab的函数来实现,三个参数分别是视频路径,步长(每隔多少帧存储一张图片)和图片存储地址。但是,matlab在使用时要注意,脚本名称和函数名称一定要相同!!!
转载地址:http://qhtki.baihongyu.com/