【MMDetection系列】MMDetection3.x版本多类目标训练查看每个类别的mAP结果（coco格式数据集）

Fate_serendipity

已于 2024-05-01 16:15:25 修改

阅读量1.5k

点赞数 11

CC 4.0 BY-SA版权

分类专栏： MMDetection 文章标签：深度学习机器学习人工智能 python 目标检测

于 2024-04-25 15:49:12 首次发布

本文链接：https://blog.csdn.net/Fate_serendipity/article/details/138189714

文章目录

一、问题
二、解决方案
总结

一、问题

使用模型对coco格式的多类目标数据集训练后，得到了训练好的模型权重文件，想要在测试阶段查看每个类别的map，网上大多数教程提供的是MMDetection2.x版本的，采用的方式为：

1.在test.py 后面加“- -options "classwise=True"

python tools/test.py \
    ${
   CONFIG_FILE} \
    ${
   CHECKPOINT_FILE} \
    [--out ${
   RESULT_FILE}] \
    --eval bbox \
    --options "classwise

最低0.47元/天解锁文章