机器视觉 OpenCV—python 图像数据集获取工具(视频取帧)
一、前言之前在做图像分类的时候,人脸识别(开源代码)的练手,数据集获取麻烦(没人愿意将自己照片给人家做数据集),于是就用自己造数据集,但是拍照拍几百张训练效果不好,也嫌麻烦,干脆就是视频取帧的方式,在这之前使用专门的软件。不过opencv自带了视频处理的API,详细代码如下:二、视频取帧代码import cv2import osimport sysinput_path = s...
·
一、前言
之前在做图像分类的时候,人脸识别(开源代码)的练手,数据集获取麻烦(没人愿意将自己照片给人家做数据集),于是就用自己造数据集,但是拍照拍几百张训练效果不好,也嫌麻烦,干脆就是视频取帧的方式,在这之前使用专门的软件。不过opencv自带了视频处理的API,详细代码如下:
二、视频取帧代码
import cv2
import os
import sys
def Video_fetching(input_path,frame_interval):
filenames = os.listdir(input_path) # 列出文件夹下所有的视频文件
video_prefix = input_path.split(os.sep)[-1] # 获取文件夹名称
frame_path = '{}_frames'.format(input_path) # 新建文件夹,名称为原名加上_frames
if not os.path.exists(frame_path):
os.mkdir(frame_path)
cap = cv2.VideoCapture() # 初始化一个VideoCapture对象
for filename in filenames:
if filename.endswith('.mp4'):
filepath = os.sep.join([input_path, filename])
cap.open(filepath) # VideoCapture::open函数可以从文件获取视频
n_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) # 获取视频帧数
for i in range(42): # 为了避免视频头几帧质量低下,黑屏或者无关等
cap.read()
for i in range(n_frames-42):
ret, frame = cap.read()
# 每隔frame_interval帧进行一次截屏操作
if i % frame_interval == 0:
imagename = '{}_{}_{:0>6d}.jpg'.format(video_prefix, filename.split('.')[0], i)
imagepath = os.sep.join([frame_path, imagename])
print('exported {}!'.format(imagepath))
cv2.imwrite(imagepath, frame)
else:
print('This file is not video')
cap.release() # 执行结束释放资源
if __name__=="__main__":
if len(sys.argv) < 2:
print ('依次传入参数:视频片段的路径(文件夹),设定每隔多少帧截取一帧int(20)')
exit(0)
Video_fetching(sys.argv[1], int(sys.argv[2]))
单个视频取帧
import cv2
def Video_fetching(input_path, save_path, frame_interval):
cap = cv2.VideoCapture() # 初始化一个VideoCapture对象
filename = input_path.split("/")[-1][:-4]
if input_path.endswith('.mp4'):
cap.open(input_path) # VideoCapture::open函数可以从文件获取视频
n_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) # 获取视频帧数
for i in range(20):
cap.read()
for i in range(n_frames-20): # 为了避免视频头几帧质量低下,黑屏或者无关等
ret, frame = cap.read()
if i % frame_interval == 0: # 每隔frame_interval帧进行一次截屏操作
imagepath = save_path + "/" + filename+"_"+str(i)+".jpg"
print('exported {}!'.format(imagepath))
try:
cv2.imwrite(imagepath, frame)
except Exception as e:
print(e)
else:
print('This file is not video')
cap.release()
if __name__ == "__main__":
input_path = "./mp4/d1e273f1a8e7575cfb95847d37558b95.mp4"
save_path = "./mp4_frames_more"
frame_interval = 10
Video_fetching(input_path, save_path, frame_interval)
更多推荐
所有评论(0)