目录
一.运行环境 ...........................................................................................2
二.AiDitron 简介 ......................................................................................2
三.常用术语 ...........................................................................................3
3.1 工程 ............................................................................................3
3.2 图片标记术语 ................................................................................4
3.3 界面相关按钮 ................................................................................6
四.模块详绅介绍 .....................................................................................8
4.1 AiDitron 对象定位模块 .....................................................................8
4.2 AiDitron 的像素检测模块 ................................................................ 11
4.3 AiDitron 的对象分类模块 ................................................................ 14
五、工具的组合使用 ................................................................................ 16
六.各模块的训练注意事项 ....................................................................... 17
6.1 对象定位模块 .............................................................................. 17
6.2 像素检测模块 .............................................................................. 18
AiDitron 训练工具使用说明
AiDitron 是基亍神经网络算法的人工智能深度学习软件,主要用亍目标物体(包括
缺陷)的定位和识别。
一.运行环境
1. CPU: Intel Core i5 (最低), Intel Core i7/Xeon (推荐)
2. GPU: Nvidia Graphic Card (CUDA compute capability ≥3.0)
推荐使用:GTX1060-1070-1080, GTX TITAN 及以上性能(显存 6G 及以上)
3. Memory: 推荐 8G 以上(根据训练样本增加,内存需要跟着增加)
4. OS: W indows7/10-64.
5. 支持图片格式:PNG、BMP、TIFF、 JPEG。
二.AiDitron 简介
AiDitron 含有三个模块,每一个模块具有各自的功能和特点,模块可以单独使用,
也可以组合使用。
模块类型
对象定位模块
像素检测模块
对象分类模块
简介
具有很好的定位能力,分类能力也比较
强。适用于具有固定形状的对象定位和分
类
可以实现像素级上的检测,适用于形状不
固定的缺陷检测
具有非常强的分类能力,适用于对类别识
别率要求很高的分类
三.常用术语
3.1 工程
工程的概念是一个检测项目需要的运行空间
示例:新建一个名为“飞机检测”的工程,其模块结构如图 3-1,其对应的在硬盘中文
件夹结构如图 3-1-2
其中 0 号文件夹为根目录,存放了导入的原始图片及其它相关信息。后面的模块节点
都是以根目录为基础进行操作的;
1、2、3 号文件夹分别对应亍 1、2、3 号模块节点的,存放了对应节点的图片及相关
信息;
1.te、2.te、3.te 为分别对应亍 1、2、3 号模块节点的训练完成模型,这是训练的最终
产物;
飞机检测.tep 存储了工程的节点关系等相关信息,名字必须不工程名相同,本例即
“飞机检测”
图 3-1-1
图 3-1-2
3.2 图片标记术语
名称
GT
对象定位模块
目标的位置以
及类别
像素检测模块
是目标的像素。
右键菜单选取 GT
即可进行标记
对象分类模块
图片的类别
丌确定区域,右
键菜单选取 Mask
即可进行标记
无此选项
Mask
丌确定区域,
标记方式同
GT,类别设置
成?即是
Mask。标记为
Mask 的图片区
域丌会参不训
练
背景
一张标记过 GT
一张标记过 GT
无此选项
的图片,除了
的图片,除了
GT 区域和
GT 区域和
Mask 区域以
Mask 区域以
外,剩下的即
外,剩下的即
使背景区域
使背景区域
Result
识别结果:识
识别结果:识
图片识别成的
别位置框+类别
别成目标的像
类别
+概率(工具中
素。左上角的
没有显示概率)
分数为聚类出
的目标区域是
目标的平均概
率
3.3 界面相关按钮
要对一个工程进行训练的基本流程:
第一次操作工程:
新建工程
导入图片
新建模型节点
剪切
标记图片
训练
测试
在已有的工程基础上进行训练的基本流程:
打开工程
标记图片
训练
测试
在硬盘上建立相关的工程文件夹。
打开以前新建的工程,并加载相关资源到工具中。
从源目录中导入需要处理的图片到工程中。如果需要测试新或者训练新
的图片,可以使用导入图片将图片导入到更目录 0 号文件夹中。
点击工具右上角黄色按钮,新建相应的模型节点。
根据上一级模型节点的图片和识别框,设置当前节点的有效识别区域,
并将图片剪切到当前节点文件夹中。注意:如果有多级模型节点相连,
导入图片以后,需要一级一级的剪切图片到相应的节点中。
图片标记:教给电脑需要识别的对象。丌同模块有丌同的标记方式,详绅参考各模型
图片的标记方法。
标记完图片以后,点击训练图片,工具会调用 GPU 进行模型的训练,训
练完毕后会得到相应模型节点的识别模型,后缀名为.te。工具在训练完
毕以后,会默认调用“测试”操作。
更具训练得到的识别模型,识别选中的图片集合,并进行识别率的统计
等相关操作。
在训练过程中,可以中途停止训练。测试过程中丌允许停止操作。
四.模块详绅介绍
4.1 AiDitron 对象定位模块
对象定位模块特点是具有良好的定位能力,分类能力也很强,适用于具有固定形
状的对象的定位和分类;
主要的操作流程如下:打开软件
工程
新建工程
导入图片
添加对象定位模块
剪切
画框标注 GT
训练。
1、打开软件所在文件夹,双击 AiDitron.exe,进入软件,界面如下: