yolov5中文标签的实现

1.将生成的.pt文件中标签改成对应的中文标签，生成新的pt文件，修改detect文件相应的路径，对应到新的pt文件。转换代码如下：

import torch

# 加载模型
w=torch.load('/home/code/yolov5-master/runs/train/exp3/weights/best.pt')

#打印所有name
print(w.get('model').names)

# 定义一个将英文单词映射到中文单词的字典
word_map = {
    'linnaeus': '云杉黄卷蛾',
    'Boerner': '松十二齿小蠹',
    'Leconte': '蠹虫',
    'coleoptera': '甲虫类',
    'Linnaeus': '蠹虫',
    'armandi': '大凝脂小蠹',
    'acuminatus': '六齿小蠹',
    # 根据需要添加更多映射
}

# 遍历列表，将每个英文单词替换为其中文对应词
for i in range(len(w.get('model').names)):
    if w.get('model').names[i] in word_map:
        w.get('model').names[i] = word_map[w.get('model').names[i]]

# 打印替换后的列表
print('替换后')
print(w.get('model').names)
#保存替换后的模型       
torch.save(w,'/home/code/yolov5-master/runs/train/exp3/weights/best_name.pt')

2.下载SimHei.ttf文件放在你的yolov5文件夹下，百度搜索SimHei.ttf下载即可
3.
将plot.py文件中的俩处地方换成SimHei.ttf的路径即可