解决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字体路径<matplotlib_path>/mpl-data/font下:

python
import matplotlib    
matplotlib.matplotlib_fname()

# /home/<user_name>/anaconda3/envs/<envs_name>/lib/python3.8/site-packages/matplotlib/mpl-data/matplotlibrc

sudo cp SimHei.ttf /home/<user_name>/anaconda3/envs/<envs_name>/lib/python3.8/site-packages/matplotlib/mpl-data/font/

5. 修改配置文件<matplotlib_path>/mpl-data/matplotlibrc:

sudo gedit <matplotlib_path>/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/<user_name>/.cache/matplotlib

rm -rf /home/<user_name>/.cache/matplotlib

重新预测,问题依然存在。

7. 修改plot.py的Annotator类的字体:

再次预测,问题解决:

  • 3
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值