当前位置:   article > 正文

yolov5推理中文显示_yolov5修改输出检测框名称

yolov5修改输出检测框名称

yolov5默认推理是中文显示,而实际项目部署过程中需要中文提示时需要对模型提示names进行更改。

1、在detect.py文件中找到names

这个位置的names是模型训练中类别信息(model.names)

{0:'person',....}

如果需要显示成中文

则直接将names重命名为:

names={0:'员工',...}

然后运行detect.py文件,如果报错:

UnicodeEncodeError: 'latin-1' codec can't encode characters in position 0-2: ordinal not in range(256)

需要下载字体:https://ultralytics.com/assets/Arial.Unicode.ttf

默认下载太慢可以点击链接下载,然后放到默认位置。

再运行文件,如果报错:AttributeError:'FreeTypeFont' object has no attribute 'getsize'(原因:版本过高)

则需要查找pillow的版本。

再次运行就可以满足要求。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/在线问答5/article/detail/797420
推荐阅读
相关标签
  

闽ICP备14008679号