更改前:
更改后:
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------------ #
然后配置后测试文件tools/eval.py
里面的参数(权重位置等),即可运行