醋醋百科网

Good Luck To You!

修改代码让YOLOv5支持中文标签_yolo格式的标签

供水管道维修材料yolov5模型的训练,实现自动识别

本人安装的版本是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


试训练一下结果

可以正常看到中文了

控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言