使用YOLO V5 训练自己的数据集

356 阅读1分钟

发布技术文章,文章内首/尾句带关键词“开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 1 天,点击查看活动详情

记得上一次使用yolo还是在大四毕业的时候,为了写毕业论文,学习了yolo的应用,当时是使用YOLO V3模型通过学习妇女阴道切片在电子显微镜下的病变照片,来实现辅助诊断的功能。

今天在公司,和同事闲聊时,发现同事正在研究通过机器视觉自动判断蓝宝石贴片的缺陷(气泡)检测,我突然想到YOLO可能能帮上忙,遂和同事要来了贴片的图片集

5e25faf3408ba942aecadbaceb35e76.jpg

图中白色雾状体即为缺陷(气泡)

话不多说,首先先做标签标注(将待检测区域标注起来)

标注我使用的是一款在线标注工具 Make Sense

  1. 点击Get Started

image.png

  1. 点击Drop images 上传待标记的所有图片

image.png

  1. 点击Object Detection(物体检测)

image.png

  1. 上传标签 (在本地建立一个txt文档,输入待检测的所有标签,一行一个,比如,要检测的有 apple banana,在txt里一行一个即可,之后选择Load labels from file,上传刚刚新建的txt文件)

image.png

  1. 标注一个区域,在右侧选择该区域的标签

image.png

  1. 导出YOLO V5 需要的标注文件(txt格式)

image.png

至此,我们获得了数据集的标注文件

image.png

接下来就需要下载YOLO V5模型源码