本人安装的版本是YOLOv5-6
以前的版本需要改动比较多,6.0版本官方添加了画中文标签的方法,给我们省了不少事儿,但是在评估图中的中文还是显示方框,还是需要修改代码才可以正常显示。
第一步 修改utils/general.py文件
with open(data, errors='ignore') as f:
修改为
with open(data, encoding="utf-8",errors='ignore') as f:
第二步 修改utils/metrics.py文件
sn.set(font_scale=1.0 if self.nc < 50 else 0.8) # for label size
修改为
sn.set(font="SimHei",font_scale=1.0 if self.nc < 50 else 0.8) # for label size
第三步 utils/plots.py文件
在plots.py 文件开头添加如下代码
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus']=False
试训练一下结果
可以正常看到中文了