基于DCGAN的图像修复技术实践与优化

1. 项目背景与核心价值

去年帮导师带本科生毕设时，发现图像修复这个选题每年都有学生选，但真正能做出实用效果的寥寥无几。传统基于OpenCV的修补方法在处理大面积缺失时，修复区域往往会出现明显的模糊和伪影。而基于DCGAN的修复方案，通过对抗生成网络的特征学习能力，能够实现更自然的图像重建效果。

这个毕设项目的核心价值在于：使用PyTorch框架搭建DCGAN网络，实现端到端的图像修复流程。相比传统方法，其优势主要体现在三个方面：

• 对任意形状缺失区域都能生成合理的语义内容
• 修复边缘过渡自然，不会产生明显的拼接痕迹
• 支持高分辨率图像修复（实测在512x512分辨率下仍能保持良好效果）

2. 技术方案设计

2.1 网络架构选择

采用DCGAN而非普通GAN的原因在于：

1. 全卷积结构更适合处理二维图像数据
2. 使用转置卷积实现上采样，避免棋盘伪影
3. 批归一化层使训练更稳定

网络结构参数配置示例：

Generator( (main): Sequential( # 输入为100维噪声向量 (0): ConvTranspose2d(100, 512, 4, 1, 0, bias=False) (1): BatchNorm2d(512) (2): ReLU(True) # 中间层省略... (6): ConvTranspose2d(64, 3, 4, 2, 1, bias=False) (7): Tanh() ) )

2.2 损失函数设计

采用复合损失函数：

• 对抗损失：LSGAN的MSE损失
• 感知损失：VGG16特征图差异
• 像素级L1损失

def loss_function(real_pred, fake_pred, output, target): adv_loss = mse_loss(fake_pred, torch.ones_like(fake_pred)) perc_loss = vgg_loss(output, target) pixel_loss = l1_loss(output, target) return 0.5*adv_loss + 0.3*perc_loss + 0.2*pixel_loss

3. 关键实现步骤

3.1 数据预处理

使用Paris StreetView数据集时需注意：

1. 将所有图像resize到256x256
2. 像素值归一化到[-1,1]区间
3. 随机生成矩形mask模拟缺失区域

transform = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(256), transforms.ToTensor(), transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)) ])

3.2 训练技巧

1. 使用Adam优化器，β1设为0.5
2. 生成器和判别器交替训练（1:1比例）
3. 初始学习率设为0.0002，每50epoch衰减一半

重要提示：判别器不要训练得太强，否则会导致梯度消失。可通过监控loss曲线判断，理想情况是两者loss值保持动态平衡。

4. 效果优化方案

4.1 后处理方法

原始输出可能存在局部色偏，建议：

1. 使用导向滤波进行边缘保持平滑
2. 在HSV空间调整色彩饱和度
3. 对修复区域进行Poisson融合

4.2 常见问题解决

问题：修复区域出现重复纹理 解决：增大感知损失的权重系数

问题：训练后期出现模式崩溃 解决：添加小批量判别(minibatch discrimination)层

问题：高分辨率图像修复质量下降 解决：采用渐进式增长训练策略

5. 项目扩展方向

实际部署时可考虑：

1. 结合边缘检测引导修复（Canny+GAN）
2. 针对特定场景微调（如人脸修复使用CelebA数据集）
3. 开发交互式修复工具（用户指定修复区域）

训练好的模型在GTX 1080Ti上处理512x512图像约需0.3秒，基本满足实时性要求。建议使用Flask封装成Web服务，方便实际应用。