本文主要是介绍YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
显示结果
更改前:
更改后:
方法
- 将
tools/eval.py
中的'--do_pr_metric'
后面改为default=True
即可打印出map值 - 原本是显示
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
里面的参数(权重位置等),即可运行
这篇关于YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!