26年WP电子信息工程 基于GAN的钢坯编号图片修复与色彩处理研究11.61-AI全日(22341字).docx

WHKJ 大学本科毕业设计 摘 要为了对工业环境下钢坯编号图像的字符磨损、偏色和过曝进行修复和处理，提出了一种基于改进 Pix2Pix 生成对抗网络的图像修复和色彩处理方法。首先在生成器的瓶颈层加入 CBAM 注意力模块，使得生成器更加关注字符，同时在生成器的输出端加入一个可学习的色彩校正模块来改善色调，其次构建了层次化的损失函数，除了对抗损失、L1 损失、感知损失外，还加入了字符笔画梯度损失来保证字符的边缘锐利，最后使用降容、Dropout 和强数据增强来避免小样本过拟合。实验结果表明，修复后的图像的客观指标和清晰度都有所提高，字符边缘锐利，色彩...