tf-faster-rcnn執行自己到資料集記錄
阿新 • • 發佈:2018-12-20
我只從遇到問題到地方開始記錄,沒有問題到地方按照作者的read me做下去即可。
1、執行到第3步,準備測試demo 時候出現問題
解決辦法:自己下載三個訓練資料集,然後解壓到tf-faster-rcnn根目錄下,並改名為VOCdevkit2007,然後放置到data資料夾中,重新執行命令即可。
2、default資料夾下缺少四個檔案
將第2步:ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default命令列做更改
改為:ln
-s ../../..
/voc_2007_trainval
+voc_2012_trainval .
/default
重新執行命令即可
3、第4步使用預訓練的restnet101模型進行測試
tf-faster-rcnn/lib/datasets/voc_eval.py檔案下121行:with open(cachefile,'w') as f 改為: with open(cachefile,'wb') as f
105行: cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile) 改為: cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile.split("/").split(".")[0])
然後進行測試:
./experiments/scripts/test_faster_rcnn.sh $GPU_ID pascal_voc_0712 res101
測試結果如下: