ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)

2026/9/16 1:28:06 拓冰建站 浏览量
YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)

显示结果

更改前:

在这里插入图片描述

更改后:
在这里插入图片描述

方法

  1. tools/eval.py中的'--do_pr_metric'后面改为default=True即可打印出map值
  2. 原本是显示0.538,变成显示为53.79,方法为👇

YOLOv6-main/yolov6/core/evaler.py中做如下更改:

第一步,找到第一个# Print results然后做如下更改:

在这里插入图片描述

                # ---------------lwd edit------------ #pf = '%-16s' + '%12i' * 2 + '%12.4g' * 5  # print formatLOGGER.info(pf % ('all', seen, nt.sum(), mp * 100, mr * 100, f1.mean(0)[AP50_F1_max_idx] * 100, map50 * 100, map * 100))# ---------------lwd edit------------ #

第二步,找到# Print results per class做如下更改:
在这里插入图片描述

                        # ---------------lwd edit------------ #LOGGER.info(pf % (model.names[c], seen, nt[c], p[i, AP50_F1_max_idx] * 100, r[i, AP50_F1_max_idx] * 100,f1[i, AP50_F1_max_idx] * 100, ap50[i] * 100, ap[i] * 100))# ---------------lwd edit------------ #

然后配置后测试文件tools/eval.py里面的参数(权重位置等),即可运行