Fast RCNN筆記3:tf-faster-rcnn 使用訓練好的fastrcnn模型測試test

tensorflow版本fastrcnn實現，原github：https://github.com/endernewton/tf-faster-rcnn

參考：1. https://github.com/CharlesShang/TFFRCNN 

　　　2. https://blog.csdn.net/char_QwQ/article/details/80980505

基本可以按照endernewton在git中來做，除了接下來1.沒有有寫的和2.需要改的部分對應原文中Train your own model

接下來主要實現訓練好的數據對Pascal voc2007數據集進行測試：

1.將數據集下載完成后解壓

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

 

tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar
tar xvf VOCdevkit_08-Jun-2007.tar

將文件夾以VOCdevkit2007命名copy到data下，形成以下目錄結構/tf-faster-rcnn/data/VOCdevkit2007/

其下應該有以下文件（也可以參照CharlesShang中用軟鏈接）：

2.如果完成了第一步demo測試那么就可以直接運行test(沒有需要下載pre-trained model參照原git中

Train your own model

1.Download pre-trained models and weights. )：

運行之前還需要修改tf-faster-rcnn/lib/datasets/voc_eval.py下105行和121行。

105：將

cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile)

注釋掉改為

cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile.split("/")[-1].split(".")[0])

 

121行w改為wb

　　

運行測試命令：

GPU_ID=0
./experiments/scripts/test_faster_rcnn.sh $GPU_ID pascal_voc_0712 res101

GPU_ID自己看用哪一塊,res101:下載的model pascal是數據集格式

3.運行結果如下：