YOLO模型初次训练体验(+实测)
1.训练目的
做一个简单的示例,本次训练的目的希望模型能够识别桌面的两个图标。(主要是方便准备数据)
2.数据准备
安装一个截图软件,在桌面不同分辨率,不同背景的情况下,随机调整两个图标的位置并截图保存。
原始图片:
先为截图批量重命名:
使用重命名工具,设置命名规则,如果不懂可以看后面的help。
我这里是全部按照序号重命名。
Rename - 批量文件重命名工具
重命名后可以看到结果:
安装下面的目录整理放入:
一般把整理的素材的五分之一到六分之一放到验证集中进行验