Faster-RCNN-tf使用訓練好的模型驗證測試集 test_net.py
阿新 • • 發佈:2018-11-03
對應原始碼地址:https://github.com/endernewton/tf-faster-rcnn
1、開啟tools目錄下的test_net.py檔案
修改(1):
parser.add_argument('--model', dest='model', help='model to test', default='../output/vgg16/voc_2007_trainval+voc_2012_trainval/default/vgg16_faster_rcnn_iter_5000.ckpt', type=str)
修改default為我們訓練完資料後的模型地址
修改(2):
parser.add_argument('--net', dest='net', help='vgg16, res50, res101, res152, mobile', default='vgg16', type=str)
修改default為我們自己使用的網路型別
修改(3):
接著執行test_net.py時可能出現以下錯誤:
pickle.dump(recs, f)
TypeError: write() argument must be str, not bytes
問題及修改請參考:https://blog.csdn.net/oMoDao1/article/details/83146740
修改後請把以下檔案刪除:
如果不刪除,將可能出現以下錯誤:
2、最後再次執行test_net.py檔案