最近需要做一个数据集,考虑到样本为视频的原因需要对文件夹下所有视频进行抽帧处理,将样本处理为图片后进行标注,以便网络进行训练。
本文代码实现的主要功能有:
1、读取到文件夹下每个视频的FPS、帧数,以及所有视频的总帧数。
2、按照指定的帧数抽帧处理,每隔n帧保存一帧。
3、按照每个抽帧的视频名称生成对应的文件夹保存图片。
4、将成功抽帧的视频名称写入txt文件,方便检查。
代码如下:
# -*- coding: utf-8 -*-
import cv2
import os
import pdb
import numpy as np
from glob2 import glob
videos_src_path = 'C:\\Users\\Zhang\\Desktop\\aaa\\buy\\' # 提取图片的视频文件夹
#筛选文件夹下MP4格式的文件
#videos = os.listdir(videos_src_path) # 用于返回指定的文件夹包含的文件或文件夹的名字的列表。
#videos = filter(lambda x: x.endswith('mp4'), videos)
dirs = os.listdir(videos_src_path) # 获取指定路径下的文件
#根据名称创建对应的文件夹
# def mkdir(path):
# folder=os.path.exists(path)
# if not folder:
# os.makedirs(path)
# print(path+"---Done---")
# else:
# print(path+"---This is the folder---")
count = 0
#数总帧数
# total_frame = 0