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，

bbox，segm 是 COCO 数据集的可选值，mAP，recall 是 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