video2frame_multiple.py & mypath.py

本文仅限作者备忘
程序调试要点备忘:这两个程序可用来对包含很多种类的数据集分帧,并将分帧后的图片按一定比例划分为train, val和test。本文以UCF101为例进行程序调试。

mypath.py

1. 修改路径

修改root_diroutput_dir,两者分别对应为原始视频文件夹(数据集)帧保存路径
【原创】视频分帧程序备忘
【原创】视频分帧程序备忘
有没有/都可,一开始以为跟
【原创】视频分帧程序备忘
创建./video_test/train有关需要有/,后来发现没有也行。

video2frame_multiple.py

1. numpy版本

2行:
【原创】视频分帧程序备忘
不能太高,不然会显示from sklearn.model_selection import train_test_split,即无法加载系统自带文件sklearn.model

2. 修改数据集标注路径

60-64行:
【原创】视频分帧程序备忘

3. 手动创建train&val&test文件夹

119-123行:
output_dir目录下,手动创建train&val&test文件夹
【原创】视频分帧程序备忘
❓未来可考虑自动创建

4. 打开数据预处理

251行:
【原创】视频分帧程序备忘
设置preprocess=True

video2frame_multiple.py的整体伪代码流程

【原创】视频分帧程序备忘

相关文章: