针对成像中直流分量变化的深度学习样本增强方法技术

本发明专利技术涉及一种针对成像中直流分量变化的深度学习样本增强方法，包括以下步骤，S1、样本选取；S2、特征提取；S3、训练图像集数据处理；S4、目标图像集数据处理；S5、网络设计；S6、网络训练；S7、迭代训练；S8、网络使用。本发明专利技术利用在不同成像环境下得到的样本图像，进行直方图统计，利用对抗神经网络训练出不同成像环境下的直方图转换网络；然后利用网络输出的直方图对原图进行直方图规定化，从而生成模拟不同成像环境下，且效果接近真实的样本图像，实现样本增强，用于设备上深度学习网络模型的训练，从而提高模型的泛化能力，无需重新训练。无需重新训练。无需重新训练。

【技术实现步骤摘要】
针对成像中直流分量变化的深度学习样本增强方法


[0001]本专利技术涉及图像视觉检测
，尤其是一种针对成像中直流分量变化的深度学习样本增强方法。

技术介绍

[0002]自动化机器视觉行业一直以来存在不同机台成像效果存在差异的问题，尤其是量产机台，每台机器可能是由不同光电人员调试并定型的，不同光电人员的经验，能力，评价标准等存在差异，使得成像效果差异更大。
[0003]这种差异直接导致在样机上的算法无法在新机台上直接复制使用，对于传统算法，可以通过参数设计，调整参数的方法完成快速复制；但对于使用到深度学习算法的设备，由于深度学习是一个端到端的过程，在使用阶段几乎没有中间参数可供调节，检出效果主要由网络设计以及训练样本及训练方法决定，无法像传统算法那样通过设计中间参数，在使用的时候调节中间参数，使算法能快速复用到其他机台上。
[0004]对于这个问题，现有的已知的解决办法都是从样本增强上着手解决，通常有以下三种方式；（1）最原始的办法是直接将图像全局各通道的灰度分别加上或者减去一个恒定值，用来模拟不同成像环境下的样本图片；（2本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种针对成像中直流分量变化的深度学习样本增强方法，其特征在于：包括以下步骤，S1、收集同一种产品在各种成像环境下的样本，作为不同的子集，分别为训练图像集与目标图像集；S2、提取训练图像集每张图像的各通道的直方图，并对直方图向量进行归一化；S3、针对每张训练图像集中的图像，生成高斯随机信号，并将这个高斯随机信号分别与该幅图像的各通道的经过归一化的直方图进行一维拼接，形成向量，作为一个训练样本，最终由获得的训练样本组成训练集；S4、提取目标图像集每张图像的各通道的直方图，并对直方图向量进行归一化，构成一个目标样本，最终由获得的目标样本组成目标集；S5、将训练集中的样本输入GAN中的生成网络，生成网络由若干个全连接层构成，其中输出层为SoftMax层，最终输出直方图向量；将生成网络输出的直方图向量输入GAN中的判别网络，其标签为0；将目标集中的样本输入GAN中的判别网络，其标签为1；S6、按照以上的GAN输入和标注，对GAN进行训练；S7、将步骤1中得到的各个子集轮流两两组合，将每个组合中的2个子集轮流作为训练集与目标集，按照步骤S2与步骤S6来继续训练同一个GAN网络，在该GAN网络上不断的重复训练，直到所有组合训练完成；S8、训练完成后，将生成网络应用于样本增强，将待增强样本图像参照步骤S3的方法，构成输入向量集，将输入向量集中的每个向量集输入到生成网络，生成一个直方图向量，将该直方图结合它对应的原样本图像，进行直方图规定化，得到生成的增强样本；循环将待增强样本图像进行该操作，每次用于拼接，来生成输入信号的高斯信号均为重新随机生成，直到生成的增强样本数达到规定数量。2.如权利要求...

【专利技术属性】
技术研发人员：都卫东，王天翔，
申请(专利权)人：征图新视江苏科技股份有限公司，
类型：发明
国别省市：