arXiv reaDer
単一画像のかすみ除去のためのGANベースの入力サイズ柔軟性モデル
A GAN-Based Input-Size Flexibility Model for Single Image Dehazing
敵対的生成ネットワーク (GAN) に基づく画像から画像への変換は、さまざまな画像復元アプリケーションで最先端のパフォーマンスを達成しています。単一画像のかすみ除去は典型的な例であり、かすみのある画像のかすみのない画像を取得することを目的としています。このホワイト ペーパーでは、単一画像のかすみ除去という困難なタスクに焦点を当てています。大気散乱モデルに基づいて、かすみのない画像を直接生成する新しいモデルが設計されています。画像のかすみ除去の主な課題は、大気散乱モデルに透過マップと大気光という 2 つのパラメータがあることです。それらをそれぞれ推定すると、誤差が蓄積され、かすみ除去の品質が損なわれます。この理由とさまざまな画像サイズを考慮して、cGANフレームワークを使用した画像から画像への変換のトレーニング段階とテスト段階の両方での入力サイズの柔軟性である、単一画像のかすみ除去用の新しい入力サイズの柔軟性条件付き敵対的生成ネットワーク(cGAN)が提案されています.シンプルで効果的な U 接続残差ネットワーク (UR-Net) を提案して、ジェネレータを結合し、空間ピラミッド プーリング (SPP) を採用してディスクリミネータを設計します。さらに、モデルはマルチロス関数でトレーニングされます。この場合、一貫性損失は、この論文で設計された新しい損失です。最後に、最先端の単一画像のかすみ除去性能を実現するために、マルチスケール cGAN 融合モデルが構築されます。提案されたモデルは、かすみ画像を入力として受け取り、かすみのない画像を直接出力します。実験結果は、提案されたモデルの有効性と効率を示しています。
Image-to-image translation based on generative adversarial network (GAN) has achieved state-of-the-art performance in various image restoration applications. Single image dehazing is a typical example, which aims to obtain the haze-free image of a haze one. This paper concentrates on the challenging task of single image dehazing. Based on the atmospheric scattering model, a novel model is designed to directly generate the haze-free image. The main challenge of image dehazing is that the atmospheric scattering model has two parameters, i.e., transmission map and atmospheric light. When they are estimated respectively, the errors will be accumulated to compromise the dehazing quality. Considering this reason and various image sizes, a novel input-size flexibility conditional generative adversarial network (cGAN) is proposed for single image dehazing, which is input-size flexibility at both training and test stages for image-to-image translation with cGAN framework. A simple and effective U-connection residual network (UR-Net) is proposed to combine the generator and adopt the spatial pyramid pooling (SPP) to design the discriminator. Moreover, the model is trained with multi-loss function, in which the consistency loss is a novel designed loss in this paper. Finally, a multi-scale cGAN fusion model is built to realize state-of-the-art single image dehazing performance. The proposed models receive a haze image as input and directly output a haze-free one. Experimental results demonstrate the effectiveness and efficiency of the proposed models.
updated: Sun Oct 09 2022 06:28:48 GMT+0000 (UTC)
published: Fri Feb 19 2021 08:27:17 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト