在前面的文章中已经详细介绍了在本机上安装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 : False2.4 删除matplotlib的缓冲目录
# python 终端获取缓存路径 import matplotlib print(matplotlib.get_cachedir()) # /root/.cache/matplotlib # 删除缓冲目录 # (shell) rm -rf /root/.cache/matplotlib2.5 重新测试,解决问题 二. 测试YOLOv5时无法显示中文标签 1. 遇到的问题
在对YOLOv5模型进行测试时,无法显示中文,导致标签乱码
2. 解决方法修改yolov5/utils/plots.py文件中的Annotator类中的font,修改成刚刚下载的SemHei.ttf的路径即可,如下图所示
然后运行detect.py文件即可正常显示中文



