1. 程式人生 > >tf-faster-rcnn執行自己到資料集記錄

tf-faster-rcnn執行自己到資料集記錄

我只從遇到問題到地方開始記錄,沒有問題到地方按照作者的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

測試結果如下: