1.一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:包括以下步骤:S1:搭建一种基于生成对抗网络的知识蒸馏数据增强方法的数据增强系统,包括输入模块、增强图像模块、网络训练模块和知识蒸馏模块;输入模块的输出端分别连接增强图像模块的输入端和网络训练模块的输入端,增强图像模块的输出端和网络训练模块的输出端分别连接知识蒸馏模块的输入端;
S2:向系统输入待增强数据集并确认待增强图像的格式;
S3:选取固定尺寸的掩模与待增强图像相乘运算,进行区域丢弃得到补丁图像;
S4:生成对抗网络对丢弃区域填补补丁,根据补丁图像的像素分布,生成一块填补区域补全补丁图像,进行再生成得到增强后图像;
S5:通过卷积神经网络对待增强数据集进行五折交叉检验训练得到神经网络分类器模型;
S6:神经网络分类器模型对增强后图像进行计算得到预测值S;
S7:进行知识蒸馏,将原始图像标签ground_truth与预测值S加权相加混和得到增强图像标签;
S8:系统输出图像形式的包括增强后图像和增强图像标签的最终结果。
2.根据权利要求1所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S2中,待增强数据集中的待增强图像为RGB三通道图;若待增强数据集中的图像为单色图、ARGB四通道图或BGR三通道图,则系统读入待增强图像后进行通道转化变为RGB三通道图。
3.根据权利要求1所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S3中,具体步骤为:S31:在待增强图像的宽的取值范围内和高的取值范围内生成一块由像素值0组成的掩模,掩模的大小为待增强图像的大小的S32:将图像与掩模进行乘法运算。
4.根据权利要求1所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S4中,具体步骤为:S41:通过生成对抗网络生成判别器;判别器包括3个卷积层和1个全连接层;
S42:通过生成对抗网络生成生成器;生成器包括编码器和解码器,编码器包括4个卷积层,解码器包括2个上采样层和3个卷积层;
S43:冻结判别器,训练生成器;
S44:当生成器的损失达到目标时,将生成器生成的填补区域输入到判别器中,解冻并训练判别器;
S45:得到与补丁图像像素分布一致的填补区域并对补丁图像进行填补,获得增强后图像。
5.根据权利要求1所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S5中,具体步骤为:S51:将待增强数据集分为训练集和验证集分别进行训练;
S52:设最小学习率 为0,最大学习率 为0.01,当前周期为Tcur,最大周期Ti的初始值为10,每当Tcur=Ti时,将Ti扩大两倍;则在训练过程中根据式(1)计算学习率调整策略:
6.根据权利要求5所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S6中,具体步骤为:设神经网络分类器模型共有j个输出,则第i个待增强图像的预测值Si为:
7.根据权利要求6所述的一种基于生成对抗网络的知识蒸馏数据增强方法,其特征在于:所述的步骤S7中,具体步骤为:设第i个增强后图像的标签为labeli,第i个原始标签值为ground_truthi,则通过式(3)得到增强图像标签为:labeli=0.7×ground_truthi+0.3×Si (3)。
8.用于权利要求1至7中任意一项所述的一种基于生成对抗网络的知识蒸馏数据增强方法的数据增强系统,其特征在于:包括输入模块、增强图像模块、网络训练模块和知识蒸馏模块;输入模块的输出端分别连接增强图像模块的输入端和网络训练模块的输入端,增强图像模块的输出端和网络训练模块的输出端分别连接知识蒸馏模块的输入端;
输入模块用于读入待增强数据集;
增强图像模块用于对待增强图像进行区域丢弃获得补丁图像,并通过生成对抗网络生成增强后图像;
网络训练模块用于对待增强数据集进行五折交叉检验训练获得神经网络分类器模型,神经网络分类器模型用于对增强后图像进行计算得到预测值;
知识蒸馏模块用于混和原始图像标签与预测值计算得到增强图像标签。
9.根据权利要求8所述的系统,其特征在于:对抗生成网络包括生成器与判别器;生成器包括编码器和解码器,编码器包括4个按信号流向依次连接的编码复合结构,每个复合结构包括按信号流向依次连接的1个用于提取特征的卷积层和1个用于分布调整特征激活值的BN层;解码器包括2个按信号流向依次连接的解码复合结构,每个解码复合结构包括按信号流向依次连接的1个用于放大特征图的上采样层和、1个卷积层和1个BN层;
判别器包括按信号流向依次连接的3个判别复合结构和1个全连接层,每个判别复合结构包括按信号流向依次连接的1个卷积层和1个BN层;卷积层的核尺寸为K、卷积层数为N、步长为S、填充宽度为P。
10.一种计算机存储介质,其特征在于:其内存储有可被计算机处理器执行的计算机程序,该计算机程序执行如权利要求1至权利要求7中任意一项所述的一种基于生成对抗网络的知识蒸馏数据增强方法。