1. 程式人生 > >faster-rcnn tensorflow windows demo測試

faster-rcnn tensorflow windows demo測試

重點參考連結:

第1步:將原始碼和需要用到的資料集先下載下來

the training, validation, test data and VOCdevkit資料集:

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

第2步:

1、將三個資料集下載解壓到同一個資料夾下:Faster-RCNN-TensorFlow-Python3.5-master\data\VOCdevkit2007

\

2、將VGG16解壓到\data\imagenet_weights\資料夾下,並將vgg_16改成vgg16

第3步:執行train.py

第4步:執行demo.py

執行demo時如果出現如下錯誤:

OSError: output\vgg16\voc_2007_trainval+voc_2012_trainval\vgg16\vgg16.ckpt.meta not found.
Did you download the proper networks from our server and place them properly?

解決辦法:

把demo.py Parse_args()函式的default改成vgg16;

再把Demo.py中最開始的NETS中的vgg16_faster_rcnn_iter_70000.ckpt改成vgg16.ckpt

之後在根目錄下根據提示新建./output/vgg16/voc_2007_trainval+voc_2012_trainval/vgg16/資料夾,

把模型訓練好以後生成的default資料夾下的這四個檔案copy到剛剛新建的資料夾下

並改名如下:

最後測試結果:

另附一張本人自己專案中用該模型訓練後細胞檢測結果圖: