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

发布时间:2024年01月12日

显示结果

更改前:

在这里插入图片描述

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

方法

  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 format
                LOGGER.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里面的参数(权重位置等),即可运行

文章来源:https://blog.csdn.net/LWD19981223/article/details/135562547
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。