打印YOLOv5训练过程中每个类别的AP

在YOLOv5的训练中,通常只显示mAP_0.5和mAP_0.5:0.95。若要实时查看每个类别的AP,需编辑val.py文件,取消第270行附近的注释,并将后续行向前缩进,这样训练时每轮都会输出各类别的AP值。
摘要由CSDN通过智能技术生成

  显示YOLOv5训练过程中单个类的AP

        YOLOv5训练状况下默认只显示mAP_0.5和mAP_0.5:0.95,想要查看最终每个类别的AP只能在训练结束后的PR曲线中查看。

        如果想在训练过程中打印,需要在val.py中270行附近修改

 if (verbose or (nc < 50 and not training)) and nc > 1 and len(stats):
        for i, c in enumerate(ap_class):
            LOGGER.info(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))

将第一行注释掉,并第二行第三行向前缩进:

    #if (verbose or (nc < 50 and not training)) and nc > 1 and len(stats):
    for i, c in enumerate(ap_class):
        LOGGER.info(pf % (names[c], seen, nt[c], p[i], r[i], ap50[i], ap[i]))

此时训练过程中每一轮都将显示每一个类别的AP。

  • 2
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值