赞
踩
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的版本。
再次运行就可以满足要求。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。