复现Faster-RCNN-TensorFlow-Python3.5代码,并用自己的数据集来训练(二)

知识共享许可协议 版权声明:署名,允许他人基于本文进行创作,且必须基于与原先许可协议相同的许可协议分发本文 (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

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/kellyroslyn/article/details/93294790