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

显示结果

更改前:

YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)_第1张图片

更改后:
YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)_第2张图片

方法

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

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

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

YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)_第3张图片

                # ---------------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做如下更改:
YOLOv6s,map值打印成两位小数(原本是显示0.538,变成显示为53.79)_第4张图片

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

你可能感兴趣的:(深度学习项目经验tips,YOLO)