文章出處

修改圖片格式

matlab代碼

其實內部一些代碼是用了rbg的fast-rcnn代碼的。

\datasets\VOCdevkit2007\VOCcode\VOCinit.m里面，查找'jpg',改成'png'

以及，\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}里面修改


文章列表


不含病毒。www.avast.com
arrow
arrow
    全站熱搜
    創作者介紹
    創作者 大師兄 的頭像
    大師兄

    IT工程師數位筆記本

    大師兄 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄