import argparse
import os
import os.path
from shutil import rmtree, move
import random
from glob import glob
import time
# For parsing commandline arguments
parser = argparse.ArgumentParser()
# parser.add_argument("--ffmpeg_dir", type=str, default="/home/lcj/ffmpeg-4.1/", required=False, help='path to ffmpeg.exe')
parser.add_argument("--ffmpeg_dir", type=str, default="/home/zrway/TempNew/SDK发布/transcoding_v1.0.5/test/", required=False, help='path to ffmpeg.exe')
parser.add_argument("--dataset", type=str, default="UCF-101",
help='specify if using "UCF-101" or "custom" video dataset')
parser.add_argument("--videos_folder", type=str, default="./UCF-101/", required=False,
help='path to the folder containing videos')
parser.add_argument("--dataset_folder", type=str, default="./ucf_clips/", required=False,
help='path to the output dataset folder')
args = parser.parse_args()
def extract_frames(videos, inDir, outDir):
cnt = 0 # videos number
for video in videos[:-1]:
# os.mkdir(os.path.join(outDir, os.path.splitext(video)[0]))
cnt += 1
# print('{} -loglevel error -i {} -vsync 0 -qscale:v 2 -f image2
create_dataset
最新推荐文章于 2022-05-23 10:00:49 发布
本文详细介绍了如何利用create_dataset工具从头构建自己的数据集,包括数据预处理、划分训练集和验证集,以及如何加载和使用这些数据进行模型训练。适合想要自定义数据集的初学者和开发者参考。
摘要由CSDN通过智能技术生成