mx-rcnn训练测试
训练:
1.数据集准备:
训练图片放在mx-rcnn/data/VOCdevkit/VOC2007/ JPEGImages 文件夹下;
xml文件放在mx-rcnn/data/VOCdevkit/VOC2007/ Annotations 文件夹下;
trainval.txt放在mx-rcnn/data/VOCdevkit/VOC2007/ImageSets//Main/文件夹下。
2.修改训练参数
在mx-rcnn/rcnn/dataset目录下打开pascal_voc.py,修改self.classes,填上具体的类名;
在mx-rcnn/rcnn目录下打开config.py,修改NUM_CLASSES,填上类的数目,SCALES也可在此修改;
3.训练
在mx-rcnn目录下输入python train_end2end.py --gpu 0即可
训练结果保存在model文件夹下
4.测试
在mx-rcnn目录下打开demo.py,修改CLASSES
将包含待测试图片的jpg文件夹放入demo文件夹下
在mx-rcnn目录下执行python demo.py --prefix model/e2e --epoch 10 --gpu 0
5.断点重训
--resume --begin_epoch 10 --end_epoch 15 --lr_step 15
posted on 2017-08-18 09:41 zhangxuyang 阅读(152) 评论(0) 收藏 举报
浙公网安备 33010602011771号