基于改进CycleGAN的浑浊水体图像增强算法研究

Turbid Water Image Enhancement Algorithm Based on Improved CycleGAN

针对循环生成对抗网络(Cycle Generative Adversarial Networks,CycleGAN)在浑浊水体图像增强中存在质量差和速度慢的问题,该文提出一种可扩展、可选择和轻量化的特征提取单元BSDK(Bottleneck Selective Dilated Kernel),并利用BSDK设计了一个新的生成器网络BSDKNet。与此同时,提出一种多尺度损失函数MLF(Multi-scale Loss Function)。在自建的浑浊水体图像增强数据集TC(Turbid and Clear)上,该文BM-CycleGAN比原始CycleGAN的精度提升3.27%,生成器网络参数降低4.15MB,运算时间减少0.107s。实验结果表明BMCycleGAN适合浑浊水体图像增强任务。

In order to solve the problem of poor quality and slow speed in turbid water image enhancement based on Cycle Generative Adversarial Networks(CycleGAN),a scalable,selective and efficient block Bottleneck Selective Dilated Kernel(BSDK)is proposed,and a new generator network BSDKNet is redesigned by stacking BSDK.At the same time,Multi-scale Loss Function(MLF)is proposed to improve the structural similarity of the clear water image and the generated clear water image.On our turbid water image enhancement dataset Turbid and Clear(TC),the classification accuracy of the proposed BM-CycleGAN is 3.27%higher than that of classical CycleGAN.The generator parameters of BM-CycleGAN is 4.15 MB lower than that of CycleGAN,and the time consuming of BM-CycleGAN is 0.107 s less than that of CycleGAN.The experimental results show that BM-CycleGAN is suitable for turbid water image enhancement.