create_dataset

本文详细介绍了如何利用create_dataset工具从头构建自己的数据集,包括数据预处理、划分训练集和验证集,以及如何加载和使用这些数据进行模型训练。适合想要自定义数据集的初学者和开发者参考。
摘要由CSDN通过智能技术生成
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
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值