目录
一、项目窗口展示
原代码和数据集地址: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