mmdetection使用现有的模型进行推理

2023-03-18,,

数据集准备:

下载链接:

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

mmdetection使用现有模型进行推理的相关教程结束。

《mmdetection使用现有的模型进行推理.doc》

下载本文的Word格式文档,以方便收藏与打印。