解决YOLOv5算法中的中文标签显示问题

在前面的文章中已经详细介绍了在本机上安装YOLOv5的教程，安装YOLOv5可参考前面的文章

YOLOv5训练自己的数据集(超详细)

本人使用的是YOLOv5-5.0版本

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

1. 遇到的问题

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

2. 解决方法

     在yolov5/utils/plots.py文件中手动添加黑体字体，如下图所示

        依然提示报错：Font family [‘sans-serif’] not found.Falling back to DejaVu Sans 

 2.1 下载SimHei.ttf文件

        链接：https://pan.baidu.com/s/18R-gXeu4xqu2ecyg36HHtQ    提取码：ceaq

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

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

# python 终端获取路径

import matplotlib    
print(matplotlib.matplotlib_fname())

#/home/xxx/anaconda3/lib/python3.8/site-packages/matplotlib/mpl-data/matplotlibrc

# 下载好的SIMHEI.TTF放到该目录下 
mv ./SimHet.ttf /home/xxx/anaconda3/lib/python3.8/site-packages/matplotlib/mpl-data/fonts/ttf/

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

# (shell)
vim /home/xxx/anaconda3/lib/python3.8/site-packages/matplotlib/mpl-data/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())
# /root/.cache/matplotlib

# 删除缓冲目录
# (shell)
rm -rf /root/.cache/matplotlib

 2.5 重新测试，解决问题

二. 测试YOLOv5时无法显示中文标签

1. 遇到的问题

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

2. 解决方法

     修改yolov5/utils/plots.py文件中的Annotator类中的font,修改成刚刚下载的SemHei.ttf的路径即可，如下图所示

        然后运行detect.py文件即可正常显示中文