MMDetection——COCO格式数据集显示单类别AP

最新推荐文章于 2024-04-15 03:15:19 发布

阿基美德

最新推荐文章于 2024-04-15 03:15:19 发布

阅读量2.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： MMDetection

本文链接：https://blog.csdn.net/X_blackbutterfly/article/details/119243938

MMDetection 专栏收录该内容

7 篇文章

订阅专栏

要显示COCO格式数据集在MMDetection中的单类别平均精度（AP），需修改mmdet/datasets/coco.py文件中的evaluate函数，将classwise参数设置为True。这将允许你查看每个类别的详细AP结果。此外，可以根据需求调整iou_thrs参数来改变IoU阈值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题

COCO格式数据集如何在MMDetection中显示单类别AP

解决方案

1. 进入mmdet/datasets/coco.py

2. 修改363行中的evaluate函数参数：classwise=True

3. iou_thrs可根据自己的情况看是否修改，(输入需是包含float数据的List）

def evaluate(self,
                 results,
                 metric='bbox',
                 logger=None,
                 jsonfile_prefix=None,
                 classwise=True,
                 proposal_nums=(100, 300, 1000),
                 iou_thrs=None,
                 metric_items=None):
        """Evaluation in COCO protocol.

        Args:
            results (list[list | tuple]): Testing results of the dataset.
            metric (str | list[str]): Metrics to be evaluated. Options are
                'bbox', 'segm', 'proposal', 'proposal_fast'.
            logger (logging.Logger | str | None): Logger used for printing
                related information during evaluation. Default: None.
            jsonfile_prefix (str | None): The prefix of json files. It includes
                the file path and the prefix of filename, e.g., "a/b/prefix".
                If not specified, a temp file will be created. Default: None.
            classwise (bool): Whether to evaluating the AP for each class.
            proposal_nums (Sequence[int]): Proposal number used for evaluating
                recalls, such as recall@100, recall@1000.
                Default: (100, 300, 1000).
            iou_thrs (Sequence[float], optional): IoU threshold used for
                evaluating recalls/mAPs. If set to a list, the average of all
                IoUs will also be computed. If not specified, [0.50, 0.55,
                0.60, 0.65, 0.70, 0.75, 0.80, 0.85, 0.90, 0.95] will be used.
                Default: None.
            metric_items (list[str] | str, optional): Metric items that will
                be returned. If not specified, ``['AR@100', 'AR@300',
                'AR@1000', 'AR_s@1000', 'AR_m@1000', 'AR_l@1000' ]`` will be
                used when ``metric=='proposal'``, ``['mAP', 'mAP_50', 'mAP_75',
                'mAP_s', 'mAP_m', 'mAP_l']`` will be used when
                ``metric=='bbox' or metric=='segm'``.

        Returns:
            dict[str, float]: COCO style evaluation metric.
        """