解决yolov5第6版预测图中文显示问题

源码：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字体路径/mpl-data/font下：

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. 修改配置文件/mpl-data/matplotlibrc：

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类的字体：

再次预测，问题解决：