你将收获
掌握如何一键标注缺陷或目标图片
掌握如何一键生成配置文件
掌握如何一键训练
适用人群
对人工智能机器视觉感兴趣的朋友们和从业者
软件环境:
Windows10 x64; cuda 10.2(可选,有显卡的需要装);
下载附件,安装智能AI训练标注平台软件
1、建一个文件夹,例如VOC2021(标注和路径最好都要 不含中文)
文件夹里复制进去你拍好的图片,比如一个OK文件,一个NG文件夹
2、打开智能AI标注训练平台
3、选择刚才刚才建的的文件夹作为项目主目录
进入NG文件夹,查看图片,并双击进入全图模式
如果图片不是jpg的格式,可以先按[改格式]按钮一键转化成jpg
点击添加标签,把NG文件夹的所有图片标记成NG
然后点下一个,继续添加直到标注完所有图片(所有缺陷位置都要标注出来)
标注所有图片后,在点前一个按钮,把所有图片都检查一遍,不要漏掉缺陷
标注完成后,点击生成按钮,一键生成所有文件
查看生成结果是否都是OK
然后点击[开始训练]按钮开始训练
如果正常的话,后台数据是不停滚动的,滚动的速度代表里的主机的性能,有显卡的主机看着是享受:)
如果没有显卡的主机,速度会很慢,最好找一台有显卡的主机,重新训练
安装cuda10.2,附件有下载
安装后可以打开nvidia的控制面板的系统信息,如下
根据不同电脑的配置经过几个小时的训练后,出现一条有拐点的曲线,说明训练完成了 ,一般20000次以上就可以
训练完成后,打开 ouput文件夹,可以看到训练好的模型文件,我们选择 xxxx.cfg,xxxx.names,其中一个.weights文件就可以,一般先选xxxx-10000.weights试试
现在用编译好的前端可执行软件来测试训练效果(可以在无GPU电脑中运行),下载程序后依次安装VC运行库,MVS驱动和应用程序
双击智能AI图像软件中exe可执行文件,打开前端软件,如果提示缺少dll文件,可能是你没有安装vc运行库或mvs驱动库,请先安装前两步
打开软件进入主界面后,点击切换到设定模式,先添加测试图像来源,我们先选择文件图像测试,测试ok后,可以把文件改成相机图像
在文件图像工具里,我们选择文件路径模式,把训练时的文件夹路径添加进来后
点确定回到设定界面时,再点击工具追加按钮,添加一个深度识别工具,点击追加
在深度识别工具里,添加刚才xxxx.cfg,xxxx.names和xxxx.weights路径到设置界面中
在标签设置界面里,设置每个标签对应的现实名,我们喜欢修改成中文名,点击执行检测按钮,可以参考执行效果
点击确定后,回到设定主界面,点击运行按钮可以测试执行效果,我们再添加OK/NG判断逻辑,在设定主界面点击IO按钮,添加CCD判定条件和综合判定条件
点击确定按钮,回到设定主界面,点击运行按钮,测试整体运行效果
测试效果OK后,保存整个项目文件到硬盘中,可以下次或在实际项目中用
总结:
本一键式智能AI标注训练平台大大降低了深度学习的入门门槛,各种水平和年龄的朋友都可以试一下,为推进人工智能在更行业的落地加把料,希望大家的支持。
不足:
本训练平台还在持续改进完善中,大家可以把最想添加或改进的功能写在留言里,我们优先考虑人气比较高的要求,尽快在软件中实现功能。近期会添加功能会有缩略图功能和批量复制标注等功能,敬请关注,谢谢
附件:
本文所有软件例程和数据集下载地址,
百度链接:https://pan.baidu.com/s/1YPjR_TPJYLmriXNVnNbgZg 提取码:52ai
坚果云下载:00-智能AI图像检测系统 - 坚果云 - 云盘|网盘|企业网盘|同步|备份|无限空间|免费网络硬盘|企业云盘
其他下载地址:[以下隐藏,回复后可见]