1. 程式人生 > >Faster-RCNN-tf使用訓練好的模型驗證測試集 test_net.py

Faster-RCNN-tf使用訓練好的模型驗證測試集 test_net.py

對應原始碼地址: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檔案