《表2 数据集组成:基于深度学习的工业自动化包装缺陷检测方法》
将采集到的原始数据分为3类:全部带包裹(合格)、含不带包裹或者包裹有间隙(缺陷)、其他。其中‘其他’指类似于图5d没有目标线盘的图。根据实际不需要对缺陷进行分类,可简化模型,将样本分为2类:包装合格、包装缺陷。其中Inception-V3的数据集为:训练集[1280张,1266张],测试集[100张,100张]。需要说明的是:包装合格的样本容易获取,训练集和测试集中的合格样本均为原样本图,包装缺陷样本相对不易获取,训练集和测试集中的缺陷样本是原200张缺陷图经数据增强后得到的1400张图片,剔除其中一些裁剪后无线盘或者线盘显示不足一半的图片后得到1366张增强后的图片,训练时将训练集的10%作为验证集;YOLO-V3的数据集是从Inception-V3的数据集中随机抽取120张图片,使用LabelImg工具对线盘位置标注生成VOC格式的xml文件,再通过脚本生产YOLO格式的txt文件,数据集结构见表2。
图表编号 | XD00129487400 严禁用于非法目的 |
---|---|
绘制时间 | 2020.04.10 |
作者 | 李建明、杨挺、王惠栋 |
绘制单位 | 天津大学、天津大学、北京工业大学 |
更多格式 | 高清、无水印(增值服务) |