修改圖片格式
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}里面修改