arXiv reaDer
単一画像の曇り除去のためのGANベースの入力サイズの柔軟性モデル
A GAN-Based Input-Size Flexibility Model for Single Image Dehazing
生成的敵対的ネットワーク(GAN)に基づく画像から画像への変換は、さまざまな画像復元アプリケーションで最先端のパフォーマンスを実現しています。単一画像の曇り除去は典型的な例であり、ヘイズのないヘイズ画像を取得することを目的としています。このホワイトペーパーでは、単一画像の曇り除去という難しい課題に焦点を当てています。大気散乱モデルに基づいて、ヘイズのない画像を直接生成する新しいモデルを設計します。画像の曇り除去の主な課題は、大気散乱モデルに2つのパラメータ、つまり透過マップと大気光があることです。それぞれを見積もると、デヘイズの質を損なうために誤差が蓄積されます。この理由とさまざまな画像サイズを考慮して、単一画像の曇り除去のための新しい入力サイズの柔軟性条件付き生成敵対的ネットワーク(cGAN)を提案します。これは、cGANフレームワークを使用した画像から画像への変換のトレーニング段階とテスト段階の両方での入力サイズの柔軟性です。 。ジェネレーターを組み合わせ、空間ピラミッドプーリング(SPP)を採用して弁別器を設計するためのシンプルで効果的なU型残余ネットワーク(UR-Net)を提案します。さらに、モデルはマルチロス関数でトレーニングされており、この論文では整合性ロスが新しい設計ロスです。最後に、マルチスケール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, we design a novel model 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 we estimate them respectively, the errors will be accumulated to compromise dehazing quality. Considering this reason and various image sizes, we propose a novel input-size flexibility conditional generative adversarial network (cGAN) for single image dehazing, which is input-size flexibility at both training and test stages for image-to-image translation with cGAN framework. We propose a simple and effective U-type residual network (UR-Net) 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. We finally build a multi-scale cGAN fusion model 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: Fri Feb 19 2021 08:27:17 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アソシエイト