YOLOv6s，map值打印成两位小数（原本是显示0.538，变成显示为53.79）

孟孟单单

于 2024-01-12 22:02:12 发布

阅读量626

点赞数 9

分类专栏：深度学习项目经验tips 文章标签： YOLO

本文链接：https://blog.csdn.net/LWD19981223/article/details/135562547

版权

深度学习项目经验tips 专栏收录该内容

40 篇文章 11 订阅

订阅专栏

这篇文章介绍了如何在YOLOv6项目中修改eval.py脚本，通过设置`--do_pr_metric`默认为True，以及在core/evaler.py中的代码调整，使得map值从0.538变为53.79，提供了详细的步骤和关键代码片段。

摘要由CSDN通过智能技术生成

显示结果

更改前：

在这里插入图片描述

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

方法

将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 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------------ #