修改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

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐