Yolo v5-7.0版本中文标签显示方法
本文详细介绍了如何在Yolo v5-7.0版本中实现中文标签的显示方法,并附带了所需字体的下载链接。通过本文的指导,您可以轻松地将Yolo v5-7.0的目标识别结果中的标签显示为中文。
前提条件
- 本文适用于Yolo v5-7.0版本,其他版本可能不适用,请自行尝试。
- 本文假设您在训练时使用的是英文标签,只是在标记层面将英文标签用中文显示。如果您已经使用中文训练,则无需继续阅读。
实现步骤
1. 查看标记源码
在Yolo v5的源代码中,标记相关的源代码位于utils
文件夹中的plots.py
文件中。打开该文件后,找到class Annotator
这个类函数,将其中的字体改为中文字体(绝对路径),并将pil=False
改为True
。
2. 修改detect代码
由于我们是在标记层面使用中文标签,因此在源代码中还需要将训练时的英文标签和中文对应起来。最简单的办法是创建一个字典,例如:
dict_label = {"Dog": "狗", "Cat": "猫"}
然后在程序写入标签的地方将标签替换掉。具体操作如下:
- 找到原始的
label
,它返回的数据格式为“Dog 0.55”,即类别+可信度。 - 使用字典将英文标签替换为中文标签,例如:
dict_label[label[:-5]]
。 - 将替换后的标签传递给
annotator.box_label
函数。
3. 字体文件
为了实现中文标签的显示,您需要下载中文字体文件,并将其路径添加到源代码中。您可以从“C:\Windows\Fonts”中选择一个中文字体,或者使用本文提供的字体链接进行下载。
总结
通过以上步骤,您可以成功在Yolo v5-7.0版本中实现中文标签的显示。如果您在操作过程中遇到任何问题,欢迎留言讨论。