版权声明:署名,允许他人基于本文进行创作,且必须基于与原先许可协议相同的许可协议分发本文 (Creative Commons)
至此,已经用voc_2007数据集训练好了faster-rcnn模型,下面尝试用自己的数据集来训练faster-rcnn模型。
一、环境配置:
Ubuntu 16.04 Linux,显卡GeForce GTX1080 Ti;
tensorflow-gpu 1.3.0,Cuda8.0,Cudnn 6.0;
python 3.5.4
二、制作数据集
把pascal_voc.py line 25行代码的类别改为自己的类别
在图片所在文件夹打开终端,然后在命令行输入下面这一行代码可以批量将图片的 .JPG格式转为 .jpg格式。
ls *.JPG|sed -r 's#(.*).JPG#mv & \1.jpg#'|bash