YOLOv5,v8中文标签显示问题

本人使用的是YOLOv5-7.0,YOLOv8的最新版本

1. 训练YOLOv5时matplotlib无法显示中文标签

数据集中的标签是中文,在训练YOLOv5,v8算法时,matplotlib库无法显示中文

2 解决方法

在yolov5/utils/plots.py文件中手动添加黑体字体,如下图所示
YOLOv5,v8中文标签显示问题_第1张图片

2.1 下载SimHei.ttf文件

链接:https://pan.baidu.com/s/1hfQ-drCBU3z6BJpHtlKcOA?pwd=4xbz
提取码:4xbz
–来自百度网盘超级会员V7的分享

2.2 将字体SimHei.ttf放在…/mpl-data/font下

首先查询出本地存放matplotlib字体的路径

# python 终端获取路径
 
import matplotlib    
print(matplotlib.matplotlib_fname())
 
#/D:\ProgramData\Anaconda3\envs\yolov5\lib\site-packages\matplotlib\mpl-data\matplotlibrc
 
# 下载好的SIMHEI.TTF放到该目录下 
D:\ProgramData\Anaconda3\envs\yolov5\lib\site-packages\matplotlib\mpl-data\fonts\ttf\

2.3 修改…/mpl-data/matplotlibrc配置文件

# 用记事本打开matplotlibrc
 
# 修改的内容
 
# 去掉前面的#  
font.family         : sans-serif   
 
# 去掉前面的#,并在冒号后面添加SimHei
font.sans-serif     : SimHei, Bitstream Vera Sans, Lucida Grande, Verdana, Geneva, Lucid, Arial, Helvetica, Avant Garde, sans-serif  
 
# 去掉前面的#,并将True改为False
axes.unicode_minus  : False
 

2.4 删除matplotlib的缓冲目录

# python 终端获取缓存路径
import matplotlib
print(matplotlib.get_cachedir())
#C:\Users\Administrator\.matplotlib
 
# 删除缓冲目录

2.5 重新测试,解决问题

YOLOv5,v8中文标签显示问题_第2张图片

3. 测试YOLOv5,v8推理时时无法显示中文标签

3.1 遇到问题

在对YOLOv5模型进行测试时,无法显示中文,导致标签乱码

3.2 解决方法

修改yolov5/utils/plots.py,v8的ultralytics/yolo/utils/plotting.py文件中的Annotator类中的font,修改成刚刚下载的SemHei.ttf的路径即可,如下图所示
YOLOv5,v8中文标签显示问题_第3张图片
然后推理检测就正常显示中文了。
YOLOv5,v8中文标签显示问题_第4张图片

你可能感兴趣的:(目标检测,yolov5模型修改,YOLO)