mmdetection计算每个类别的AP50

修改mmdet/datasets/coco.py的classwise与 iou_thrs
    def evaluate(self,
                 results,
                 metric='bbox',
                 logger=None,
                 jsonfile_prefix=None,
                 classwise=True,#False
                 proposal_nums=(100, 300, 1000),
                 iou_thrs=[0.5],
                 metric_items=None):

直接运行测试指令./tools/dist_test.sh ./HITUAV/faster_rcnn_r50_fpn_1x_coco.py ./HITUAV/latest.pth 2(你的显卡数) --eval bbox

你可能感兴趣的:(mmdetection,Linux,Window,目标检测)