A Cascaded Convolutional Neural Network for Single Image Dehazing

1 模型设计

网络模型结构设计

post-processing:对网络估计出的传播图又用了导向滤波器进行后续的优化

2 训练

  • 数据集 NYU-V2 Depth dataset
    前提:(1)全局大气光B服从[0.7.1]均匀分布;(2)大气散射系数从[0.6,2.8]随机选择;(3)针对每个无雾清晰图像以及对应的深度图,我们随机选择5个全局大气感和大气散射系数,人工合成有雾图像;(4)所有样本采样到207x154的大小
    训练集:1300张 x 5
    验证集:101张 x 5

  • 网络参数
    权重:高斯初始化
    学习率:0.001 动量参数:0.9
    batch size:32
    损失函数:大气光估计子网络的损失函数是:SSIM误差;传播图估计子网络的损失函数是:均方误差;因此,整个网络的损失函数上面两个误差之和。
    优化方法:Adam
    框架:Tensorflow

  • 对比实验
    卷积核个数的实验比较:整个网络的卷积核个数(特征图个数)候选:8、16、32、64
    卷积核大小的实验比较:整个网络的卷积核大小(局部感受野)候选:3x3、5x5、7x7
    共享层网络深度的实验比较:共享层网络深度候选:3、4、7、9
    传输子网络深度的实验比较:传输层子网络深度候选:2DB、3DB、5DB(dense block)

3 测试

  • 合成图像测试
    比较内容:
    1、定量比较:MSE、PSNR、SSIM、Run time
    2、定性比较:显示估计出的传播图以及去雾后的彩色图
    测试集:未说明,看图应该也是NYU Depth dataset里的

  • 真实图像测试
    比较内容:传播图的比较以及去雾后彩色图的比较

猜你喜欢

转载自blog.csdn.net/qq_31425127/article/details/80334360