赞
踩
1、该方法是以中文标签进行训练与验证,请确保.yaml文件的标签是中文。
2、修改utils文件夹中的plots.py文件,在文件前面添加如下代码:
- plt.rcParams['font.sans-serif'] = ['SimHei']
- plt.rcParams['axes.unicode_minus'] = False
结果如下图所示:
3、继续在utils文件夹中的plots.py文件中找到class Annotator这个类函数,将其中的字体改为中文字体(绝对路径),再将pil=False改为Ture,建议将字体文件直接放在主目录下。
4、在utils/metrics.py文件前面添加如下代码:
- plt.rcParams['font.sans-serif'] = ['SimHei']
- plt.rcParams['axes.unicode_minus'] = False
结果如下图所示:
5、在utils/general.py文件修改def yaml_load函数,修改后如下所示:
- def yaml_load(file='data.yaml'):
- # Single-line safe yaml loading
- with open(file, errors='ignore', encoding='UTF-8',) as f:
- return yaml.safe_load(f)
'运行
6、修改train.by文件,具体如下图所示:
- if opt_yaml.is_file():
- with open(opt_yaml, errors='ignore', encoding='UTF-8') as f:
- d = yaml.safe_load(f)
7、上述步骤修改完成以后,就可以进行运行train.by文件进行训练了,当完成了用中文标签进行训练的过程后,可查看权重模型是否符合预期。若符合,则用detect.by文件进行检测。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。