[发明专利]一种基于生成式对抗网络的桥梁裂缝图像生成模型

权利要求书

1.一种基于生成式对抗网络的桥梁裂缝图像生成模型，其特征在于：包括判别子模型和生成子模型，所述判别子模型采用六层卷积层，且顺次的第一至第五层卷积层每层卷积层均采用大小为5*5的卷积核，第六层卷积层采用大小为1*1的卷积核；

所述生成子模型包括五层反卷积层，且每层反卷积层均采用大小为5*5的卷积核；

所述判别子模型的第五层卷积层和第六层卷积层后分别接入一批量归一化层；

所述生成子模型的噪声向量维数为32768；

其中，桥梁裂缝图像为利用图像采集设备采集不同背景纹理、不同材质且不含落叶的桥梁裂缝图片；

该模型为桥梁裂缝图像生成模型；

具体步骤如下：

（1）利用图像采集设备采集不同背景纹理、不同材质且不含落叶的桥梁裂缝图片，采集的图片总数为10235张，并且将所有的图片调整为128*128大小，将统一大小的图片进行90°、180°、270°的翻转，共得到30705张裂缝图片；

（2）设置真实图片的数据标签为1，生成器生成图片的数据标签为0，并设置桥梁裂缝图像生成模型参数epoch=25，batchsize =64；

（3）将裂缝图片输入到桥梁裂缝图像生成模型中的判别模型中进行特征的学习，并输出判别结果，0对应假，1对应真；

（4）构造值在[-1，1]区间内随机分布的100维随机向量Z，其中每一个随机向量的维度为32768；

（5）根据图片的数据标签与判别器输出结果做差，生成器依据计算得出的差值作为损失在进行反向回传，并采用adam优化算法对随机向量Z的分布进行调整；

（6）采用训练一次判别网络后训练两次生成网络的方法进行交叉训练；

（7）不断的重复步骤（3）至步骤（6），直至25个epoch后完成迭代，生成桥梁裂缝图像。