【faster-rcnn】训练自己的数据——修改图片格式、类别


修改图片格式

matlab代码

其实内部一些代码是用了rbg的fast-rcnn代码的。

<faster-rcnn_root>\datasets\VOCdevkit2007\VOCcode\VOCinit.m里面,查找'jpg',改成'png'

以及,<faster-rcnn_root>\imdb\imdb_from_voc.m里面,查找imdb.extension,修改它的值为'png'

python代码

py-faster-rcnn/lib/datasets/pascal_voc.py

找到self._image_ext='.jpg' 换成'.png'

修改label类别

python代码

py-faster-rcnn/lib/datasets/pascal_voc.py

找到self._classes_,保留'__background__',剩余的类别按需修改

同时记得修改solver文件中的参数。在py-faster-rcnn/models/pascal_voc/你的模型名/faster_rcnn_end2end/{train.prototxt, test.prototxt}里面修改


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM