近年、大気散乱モデル(ASM)に基づく単一画像の曇り除去ディープモデルが目覚ましい成果を上げています。しかし、これらのモデルの曇り除去出力は、カラーシフトの影響を受けます。 ASMモデルを分析すると、大気光係数(ALF)がスカラーとして設定されており、ALFが画像全体で一定であることを示しています。ただし、実世界で撮影された画像の場合、照明は画像全体に均一に分散されないため、モデルの不一致が発生し、ASMを使用したディープモデルのカラーシフトが発生する可能性があります。このことを念頭に置いて、この研究では、最初に、複雑な照明条件下で撮影されたぼんやりとした画像の画像モデリングを改善するために、新しい不均一大気散乱モデル(NH-ASM)を提案します。次に、新しいU-Netベースのフロントホワイトバランスモジュール(FWB-Module)は、大気光の推定によって曇り除去の結果を生成する前に、カラーシフトを修正するように専用に設計されています。第三に、新しいFWB損失は、FWBモジュールをトレーニングするために革新的に開発されており、カラーシフトにペナルティを課します。最終的に、NH-ASMとフロントホワイトバランス技術に基づいて、FWB-Netと呼ばれるエンドツーエンドのCNNベースのカラーシフト抑制デヘイズネットワークが開発されました。実験結果は、合成画像と実世界の画像の両方での曇り除去に対する提案されたFWB-Netの有効性と優位性を示しています。
In recent years, single image dehazing deep models based on Atmospheric Scattering Model (ASM) have achieved remarkable results. But the dehazing outputs of those models suffer from color shift. Analyzing the ASM model shows that the atmospheric light factor (ALF) is set as a scalar which indicates ALF is constant for whole image. However, for images taken in real-world, the illumination is not uniformly distributed over whole image which brings model mismatch and possibly results in color shift of the deep models using ASM. Bearing this in mind, in this study, first, a new non-homogeneous atmospheric scattering model (NH-ASM) is proposed for improving image modeling of hazy images taken under complex illumination conditions. Second, a new U-Net based front white balance module (FWB-Module) is dedicatedly designed to correct color shift before generating dehazing result via atmospheric light estimation. Third, a new FWB loss is innovatively developed for training FWB-Module, which imposes penalty on color shift. In the end, based on NH-ASM and front white balance technology, an end-to-end CNN-based color-shift-restraining dehazing network is developed, termed as FWB-Net. Experimental results demonstrate the effectiveness and superiority of our proposed FWB-Net for dehazing on both synthetic and real-world images.