源码:GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
在训练完成后,使用detect.py预测图片时,输出结果不显示标签信息:
希望得到的结果为:
通过阅读detect.py代码得到,画框对应的代码为:
找到annotator的定义位置,有:
到plots.py中,找到Annotator下的box_label方法:
经测试发现,确实self.draw.text无法显示中文内容,但英文内容可以正常显示。
参考:解决YOLOv5算法中的中文标签显示问题_AI小白一枚的博客-CSDN博客_yolov5显示中文标签
1. 修改utils/genneral.py 489行,和utils/metrics.py 196行:
2. 下载SimHei.ttf字体,保存在yolo5/路径下。
3. 在plot.py加入:
- plt.rcParams['font.sans-serif']=['SimHei']
- plt.rcParams['axes.unicode_minus'] = False
4. 将SimHei.ttf复制到matplotlib字体路径
- python
- import matplotlib
- matplotlib.matplotlib_fname()
-
- # /home/
/anaconda3/envs//lib/python3.8/site-packages/matplotlib/mpl-data/matplotlibrc -
- sudo cp SimHei.ttf /home/
/anaconda3/envs//lib/python3.8/site-packages/matplotlib/mpl-data/font/
5. 修改配置文件
- sudo gedit
/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
6. 删除matplotlib的缓冲目录:
- python
- import matplotlib
- matplotlib.get_cachedir()
-
- # /home/
/.cache/matplotlib -
- rm -rf /home/
/.cache/matplotlib
重新预测,问题依然存在。
7. 修改plot.py的Annotator类的字体:
再次预测,问题解决: