mmdetection使用現有的模型進行推理


數據集准備:

下載鏈接:

COCO:https://cocodataset.org/

VOC:http://host.robots.ox.ac.uk/pascal/VOC/

cityscapes:https://www.cityscapes-dataset.com/

 

單GPU測試

python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] [--show]

可選參數:

RESULT_FILE: 結果文件名稱,需以.pkl 形式存儲。如果沒有聲明,則不將結果存儲到文件。

EVAL_METRICS: 需要測試的度量指標。可選值是取決於數據集的,比如 proposal_fast,proposal,

bboxsegm COCO 數據集的可選值,mAPrecall Pascal VOC 數據集的可選值。Cityscapes

據集可以測試 cityscapes 和所有 COCO 數據集支持的度量指標。

--show: 如果開啟,檢測結果將被繪制在圖像上,以一個新窗口的形式展示。它只適用於單 GPU 的測

試,是用於調試和可視化的。請確保使用此功能時,你的 GUI 可以在環境中打開。否則,你可能會遇

到這么一個錯誤 cannot connect to X server

--show-dir: 如果指明,檢測結果將會被繪制在圖像上並保存到指定目錄。它只適用於單 GPU 的測

試,是用於調試和可視化的。即使你的環境中沒有 GUI,這個選項也可使用。

--show-score-thr: 如果指明,得分低於此閾值的檢測結果將會被移除。

--cfg-options: 如果指明,這里的鍵值對將會被合並到配置文件中。

--eval-options: 如果指明,這里的鍵值對將會作為字典參數被傳入 dataset.evaluation()

數中,僅在測試階段使用。

 

樣例

測試Faster R-CNN 並可視化結果(COCO)

python tools/test.py \
configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py \
checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth \
--show

checkpoints需要手動下載,下載步驟:https://www.cnblogs.com/xxmrecord/p/15778229.html

每執行一張圖像的測試,都會將結果顯示出來。叉掉就會顯示下一張。

 

測試Faster R-CNN並保存測試結果(COCO)

python tools/test.py \
configs/faster_rcnn/faster_rcnn_r50_fpn_1x.py \
checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth \
--show-dir faster_rcnn_r50_fpn_1x_results

測試Faster R-CNN,不保存測試結果(VOC)

python tools/test.py \
configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc.py \
checkpoints/faster_rcnn_r50_fpn_1x_voc0712_20200624-c9895d40.pth \
--eval mAP

注意配置文件和checkpoints和上面示例的區別。

實踐

要先下載對應得checkpoints文件,步驟:https://www.cnblogs.com/xxmrecord/p/15778229.html

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM