【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