YOLO添加中文标签

最新推荐文章于 2024-08-21 00:30:28 发布

SherryGo

最新推荐文章于 2024-08-21 00:30:28 发布

阅读量4.7k

点赞数 2

分类专栏：计算机视觉文章标签：本土化 YOLO

本文链接：https://blog.csdn.net/xueruixuan/article/details/78840846

版权

本文介绍了如何将YOLO的物体检测结果从英文转化为中文，包括在遇到问题时的解决步骤，如修改data/labels/make_labels.py，解决convert命令找不到和权限问题，以及修改image.c并重新编译YOLO，最终成功生成中文类别标签。

摘要由CSDN通过智能技术生成

YOLO默认输出检测物体的英文名称，我们需要显示本土化，本来以为只需要在代码添加label的地方汉化就可以，后发现不会有效果。
参考YOLO应用化之添加中文支持
可能是点背，我遇到的问题要比上边提到的多，故我又写了篇博客记录我的心酸历程，希望能帮到更多人。
修改步骤如下

1、修改data/labels/make_labels.py

# -*- coding: utf-8 -*-    
import os   

#l=[]
#with open("coco.names") as list_in:
#    for line in list_in:
#        l.append(line)

l=["人","自行车","车","摩托车","飞机","大巴","火车","卡车","船","交通灯","消防栓","停止标识","停车计时器","长凳","鸟","猫","狗","马","羊","牛","大象","熊","斑马","长颈鹿","背包","伞","手提包","领带","手提箱","飞盘","雪橇","滑雪板","体育用球","风筝","棒球棒","棒球手套","滑板","冲浪板","网球拍","瓶子","红酒杯","杯子","叉子","小刀"