基于改进生成对抗网络的图像数据增强方法 |
| |
引用本文: | 詹燕,胡蝶,汤洪涛,鲁建厦,谭健,刘长睿.基于改进生成对抗网络的图像数据增强方法[J].浙江大学学报(自然科学版 ),2023(10):1998-2010. |
| |
作者姓名: | 詹燕 胡蝶 汤洪涛 鲁建厦 谭健 刘长睿 |
| |
作者单位: | 浙江工业大学机械工程学院 |
| |
摘 要: | 为了提高机器学习模型的精确度,提出基于数据分布拟合、生成式对抗神经网络和图像超分辨率重建的图像数据增强方法.该方法将最大似然估计和采样算法生成的符合原始数据分布的二维噪声用于对抗训练,克服了在生成模型中传统图像噪声输入随意的问题;采用逐层训练方式生成高分辨率图像,改进高分辨率图像映射困难、参数冗余的缺点.以轴承滚子表面灰度图像数据增强为例,验证所提方法的有效性.研究结果表明,所提方法生成的图像质量更优,相比传统方法生成的图像峰值信噪比提高13.07%,结构相似性提高32.40%,弗雷歇初始距离降低37.58%,且数据增强后的模型平均精确度提升7.89%.
|
关 键 词: | 图像数据增强 分布拟合 采样算法 生成式对抗网络 图像超分辨率重建 |
|
|