用py-faster-rcnn训练自己的VOC数据集
2017-10-07 09:54
435 查看
数据集的制作遵循博主http://blog.csdn.net/gaohuazhao/article/details/60871886的规则,只需要四个步骤。
(1)用acdsee给图片重命名。比如说 000001.jpg.
(2)用labelimg给图片标注
(3)替换VOC2007里面的annonations, imageset, JPEG之类,其中第二个文件夹的main.py需要四个txt文件,第一个文件夹需要XML文件,labelimg可以生成。第三个文件夹放的就是命名好的图片
(4)更改几个配置文件,就是faster-rcnn设置的是21个类,需要在几个配置文件中改成自己的类别数+1.
然后就是训练。, 有很多人写的是利用命令行:experiments/scripts/faster_rcnn_end2end.sh experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc来运行,但是容易出现错误(F0317 21:26:39.516222 13403 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out
of memory),有人说需要更改faster-rcnn-root/lib/fast_rcnn/config.py中的batch_size,但是我看了另一个人回复的,使用
命令行:experiments/scripts/faster_rcnn_end2end.sh 0 ZF pascal_voc,利用end2end模式更省内存。
现在还在迭代。估计没啥问题了
(1)用acdsee给图片重命名。比如说 000001.jpg.
(2)用labelimg给图片标注
(3)替换VOC2007里面的annonations, imageset, JPEG之类,其中第二个文件夹的main.py需要四个txt文件,第一个文件夹需要XML文件,labelimg可以生成。第三个文件夹放的就是命名好的图片
(4)更改几个配置文件,就是faster-rcnn设置的是21个类,需要在几个配置文件中改成自己的类别数+1.
然后就是训练。, 有很多人写的是利用命令行:experiments/scripts/faster_rcnn_end2end.sh experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc来运行,但是容易出现错误(F0317 21:26:39.516222 13403 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out
of memory),有人说需要更改faster-rcnn-root/lib/fast_rcnn/config.py中的batch_size,但是我看了另一个人回复的,使用
命令行:experiments/scripts/faster_rcnn_end2end.sh 0 ZF pascal_voc,利用end2end模式更省内存。
现在还在迭代。估计没啥问题了
相关文章推荐
- py-faster-rcnn 训练自己的数据
- py-faster-rcnn(caffe)配置和运行,训练自己的数据
- py-faster-rcnn训练自己的数据
- py-faster-rcnn用自己的数据训练模型
- py-faster-rcnn用自己的数据训练模型
- py-faster-rcnn 安装、测试和训练自己的数据
- 使用py-faster-rcnn训练自己的数据
- 【py-faster-rcnn】【训练自己数据】需要修改的参数小记
- 使用faster rcnn训练自己的数据(py-faster-rcnn )
- faster rcnn训练自己地数据时遇到地问题
- Ubuntu16.04 caffe py-faster-rcnn安装以及训练自己的数据
- faster rcnn训练自己地数据时遇到地问题
- py-faster-rcnn + ZF 实现自己的数据训练与检测(二)
- 使用py-faster-rcnn训练自己的数据集
- py-faster-rcnn + ZF 实现自己的数据训练与检测(二)
- faster rcnn训练自己地数据时遇到地问题
- DL开源框架Caffe | 目标检测Faster-rcnn训练自己数据问题整理
- Deep Learning---py-faster-rcnn基于PASCAL VOC数据集训练模型
- py-faster-rcnn + ZF 实现自己的数据训练与检测(一)
- py-faster-rcnn+VGG_CNN_M_1024训练自己数据需要修改的参数小记