发布技术文章,文章内首/尾句带关键词“开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 1 天,点击查看活动详情”
记得上一次使用yolo还是在大四毕业的时候,为了写毕业论文,学习了yolo的应用,当时是使用YOLO V3模型通过学习妇女阴道切片在电子显微镜下的病变照片,来实现辅助诊断的功能。
今天在公司,和同事闲聊时,发现同事正在研究通过机器视觉自动判断蓝宝石贴片的缺陷(气泡)检测,我突然想到YOLO可能能帮上忙,遂和同事要来了贴片的图片集
图中白色雾状体即为缺陷(气泡)
话不多说,首先先做标签标注(将待检测区域标注起来)
标注我使用的是一款在线标注工具 Make Sense
- 点击Get Started
- 点击Drop images 上传待标记的所有图片
- 点击Object Detection(物体检测)
- 上传标签 (在本地建立一个txt文档,输入待检测的所有标签,一行一个,比如,要检测的有 apple banana,在txt里一行一个即可,之后选择Load labels from file,上传刚刚新建的txt文件)
- 标注一个区域,在右侧选择该区域的标签
- 导出YOLO V5 需要的标注文件(txt格式)
至此,我们获得了数据集的标注文件
接下来就需要下载YOLO V5模型源码