深度学习自学（十二）：关键点数据集处理－300VW

一、300VW关键点数据集

最近训练关键点模型缺少数据，所以整理了下300vw的视频数据，整理成：文件名＋检测框＋关键点的多标签数据。

数据路径：https://yun.baidu.com/s/1boROHcj

300vw数据为视频格式，需要处理成单帧图片并跟每个关键点pts文件对应。

处理视频数据采用FFMPEG解码器解码成单帧数据。

（１）单个视频命令

在ubutun终端中输入:

ffmpeg -i vid.avi -qscale:v 1 -f image2 image/%06d.jpg

即可将视频中的每一帧提取成为单独的图像文件，存放在当前目录的image文件夹下，文件名取自帧编号。

-i是指定输入文件，-qscale:v是指定输出的图片质量参数，-f为输出文件的模式。

如果不指定-qscale:v的话FFmpeg会使用默认参数输出，图片质量会变得很差。

（２）批处理

由于300-VW下的114个avi视频文件文件名是相同的，可以写一个简单的Shell脚本来批量将avi视频中的帧提取出来。

在对应的解压文件夹下面新建read_video.sh，将下面的命令粘贴进去，然后运行sudo sh ./read_video.sh，会在每个是文件夹生

成相应的image文件夹，里面存的的单帧图片跟ｐｔｓ文件一一对应

for dir in $(ls -d */);
do 
    cd $dir;
    mkdir image;
    ffmpeg -i vid.avi -qscale:v 1 -f image2 image/%06d.jpg;
    cd ..;
done

（３）文件重命名

因为每个文件夹下面的pts文件和图片文件名字都相同，所以需要把每个文件夹下面的数据加上对应文件夹的名字：

300vw-001-000001.pts   300vw-001-000001.jpg，这样文件合并的时候不会出现重名。建议最好三个到四个文件夹合并为一个

part进行处理，这样数据处理会快很多。