目录

 

一、项目窗口展示

二、数据集准备

1、labelimg数据标注

2、中医舌象数据集

3、数据集展示


 

一、项目窗口展示

原代码和数据集地址:https://m.tb.cn/h.g8V294l?tk=ssHf3cwymML 

二、数据集准备

        yolov5的数据集需要一个images(图像)文件夹和一个labels(标签)文件夹,分别用于存储图像数据和标签数据。如果你要训练的数据集无法在网络上找到对应的yolo格式的标签数据,那么就需要你自己对图像数据进行标注操作。

1、labelimg数据标注

这时候需要用到labelimg软件:

labelimg下载:

pip install labelimg

由于我已经下载好,所以显示如下:

labelimg启动:

在cmd窗口下输入labelimg:

正常情况下是可以启动的:启动界面如下:

左侧工具栏Open Dir 打开需要图像标注的图像存在的文件夹;Change Save Dir打开标签存放的文件夹。生成的标签是一个名称和原图像一样的txt文件,第一个数字是label对应的数字,后四个数字是识别框的坐标。

最关键的是标注格式需要改成YOLO格式,不然会导致你的yolov5模型识别不了标签:

如上图所示。

全部设置好就可以开始标注你的图像数据了。

PS:yolov5模型识别不了中文路径,所以存放文件夹不能包含中文,包括你的图像尽量用数字表明。

2、中医舌象数据集

现在我的数据集将我的数据集作为例子展示。

首先images和labels文件夹存储在yolov5项目的data文件夹下

其中images和labels文件夹下有train、val、test文件夹作为训练集、验证集和测试集。

如图,images和labels下的三个文件夹中的名称应当是相互对应的,如果没用相互对应,就会出现报错和导致训练分数为0。

3、数据集展示

原代码和数据集地址:https://m.tb.cn/h.g8V294l?tk=ssHf3cwymML 

 

 

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐