噪声图像生成模型训练制造技术

本发明专利技术涉及图像处理领域，提供了一种噪声图像生成模型训练

【技术实现步骤摘要】
噪声图像生成模型训练、去噪模型训练及去噪方法


[0001]本专利技术涉及图像处理领域，尤其涉及一种噪声图像生成模型训练
、
去噪模型训练及去噪方法
。

技术介绍

[0002]近年来，巡检机器人已逐步应用于数据中心巡检任务，有效解决了传统人工巡检效率低
、
成本高等问题，提升了智能运维管理水平
。
然而，数据中心内大量电子设备及高低压输电线路产生电磁干扰，导致机器人成像设备采集的图像受到不同程度噪声影响，引起图像边缘及细节信息缺失
。
此情况会降低后续视觉识别与定位任务的识别率与定位精度
。
因此，为提升机器人视觉系统抗干扰能力，采用图像去噪预处理以提高图像质量显得尤为重要
。
[0003]现有技术中，利用
N2N(Noise2Noise)
模型对噪声图像进行去噪
。
但是在训练过程中，
N2N
模型需要输入成对的巡检物体的噪声图像，这两幅图像需要满足两个要求：来自同一幅干净图像；噪声图像中的噪声本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种噪声图像生成模型训练方法，其特征在于，第一噪声图像生成模型包括第一生成器
、
去噪器
、
第一判别器和合成模块，所述方法包括：获取随机数据和实际噪声图像；将所述随机数据输入至所述第一生成器中，得到模拟噪声；将所述实际噪声图像输入至所述去噪器中，得到去噪图像；将所述模拟噪声和所述去噪图像输入至所述合成模块，得到合成噪声图像；将所述合成噪声图像和所述实际噪声图像输入至所述第一判别器中，得到第一判别结果；根据所述第一判别结果，对所述第一噪声图像生成模型中的第一生成器
、
去噪器和第一判别器进行训练，得到第二噪声图像生成模型，所述第二噪声图像生成模型用于合成噪声图像
。2.
根据权利要求1所述的方法，其特征在于，所述第一噪声图像生成模型还包括第二生成器和第二判别器，所述根据所述第一判别结果，对所述第一噪声图像生成模型中的第一生成器
、
去噪器和第一判别器进行训练，得到第二噪声图像生成模型，包括：根据所述实际噪声图像，获取实际噪声；将所述模拟噪声和所述实际噪声输入至所述第二判别器中，得到第二判别结果；将所述去噪图像输入至所述第二生成器中，得到模拟噪声图像；根据所述第一判别结果
、
所述第二判别结果
、
所述模拟噪声图像和所述实际噪声图像，计算第一损失函数值；对所述第一生成器中的参数
、
所述去噪器中的参数
、
所述第二生成器中的参数
、
所述第一判别器中的参数和所述第二判别器中的参数进行调整，并重新计算所述第一损失函数值，直到所述第一损失函数值不再下降，得到第二噪声图像生成模型
。3.
根据权利要求2所述的方法，其特征在于，根据所述第一判别结果
、
所述第二判别结果
、
所述模拟噪声图像和所述实际噪声图像，计算第一损失函数值，包括：根据所述第一判别结果，计算第二损失函数值；根据所述第二判别结果，计算第三损失函数值；根据所述实际噪声图像和所述模拟噪声图像，计算第四损失函数值；根据所述第二损失函数值
、
所述第三损失函数值和所述第四损失函数值，计算所述第一损失函数值
。4.
根据权利要求3所述的方法，其特征在于，根据所述第二损失函数值
、
所述第三损失函数值和所述第四损失函数值，计算所述第一损失函数值，包括：将所述第二损失函数值
、
所述第三损失函数值和所述第四损失函数值加权求和，得到所述第一损失函数值，所述第二损失函数值的权重值
、
所述第三损失函数值的权重值
、
所述第四损失函数值的权重值依次减小
。5.
根据权利要求2所述的方法，其特征在于，根据所述实际噪声图像，获取实际噪声，包括：根据所述实际噪声图像，得到多个图像块；对各所述图像块进行二维离散小波变换，得到各所述图像块对应的尺度系数图像和小波系数图像；
将尺度系数图像和小波系数图像满足预设条件的图像块作为噪声区域；将所述噪声区域中的各像素与所述噪声区域中的平均像素作差，得到所述实际噪声
。6.
根据权利要求1所述的方法，其特征在于，所述去噪器包括至少一个反转瓶颈块，所述反转瓶颈块中包括可塑卷积层，所述可塑卷积层包括深度可分离卷积层，所述将所述实际噪声图像输入至所述去噪器中，得到去噪图像，包括：根据所述实际噪声图像和所述深度可分离卷积层，得到第一特征图；根据所述第一特征图和所述实际噪声图像，得到所述去噪图像
。7.
根据权利要求6所述的方法，其特征在于，所述可塑卷积层中还包括池化层
、
预测器网络
、
上采样层和动态卷积层，所述根据所述实际噪声图像和所述深度可分离卷积层，得到第一特征图，包括：对所述实际噪声图像进行混洗操作，得到操作后的实际噪声图像；将所述操作后的实际噪声图像输入至所述池化层，得到第二特征图；将所述第二特征图输入至所述预测器网络，得到第三特征图；将所述第三特征图输入至所述深度可分离卷积层，得到多个第一滤波器；将各所述第一滤波器输入至所述上采样层，得到多个第二滤波器；将各所述第二滤波器和所述操作后的实际噪声图像输入至所述动态卷积层，得到...

【专利技术属性】
技术研发人员：李德龙，金和平，罗惠恒，吴文杰，王景晗，周超辉，张晓萌，
申请(专利权)人：中国长江三峡集团有限公司，
类型：发明
国别省市：