arXiv reaDer
雨空間の対照的な制約を伴う単一画像のドレインのための対になっていない敵対的学習
Unpaired Adversarial Learning for Single Image Deraining with Rain-Space Contrastive Constraints
ペアの合成データに依存すると、実際のアプリケーションでの一般性とスケーラビリティが制限されることが多いため、ペアのない情報を使用したディープラーニングベースの単一画像ドレイン(SID)は非常に重要です。ただし、雨と雨のない画像間のドメイン知識は非対称であるため、SIDタスクで不対の敵対的学習とサイクル整合性制約を直接使用することは、雨の入力からクリーンな出力までの基本的な関係を学習するには不十分であることに気付きました。このような制限に対処するために、CDR-GANという名前のGANフレームワークで、対になっていないエグザンプラの相互特性を対照的な学習方法で探索する、効果的な対になっていないSIDメソッドを開発します。提案された方法は、主に双方向翻訳ブランチ(BTB)と対照ガイダンスブランチ(CGB)の2つの協調ブランチで構成されています。具体的には、BTBは、敵対的一貫性の循環アーキテクチャを最大限に活用して、潜在的な機能の分布を活用し、双方向マッピングを装備することで2つのドメイン間の転送機能をガイドします。同時に、CGBは、雨の除去と画像の復元をより適切に支援するために、類似の特徴分布をより近くに促し、非類似の特徴をさらに遠ざけることによって、雨空間へのさまざまな模範の埋め込みを暗黙的に制限します。トレーニング中に、提案されたCDR-GANをさらに制約するために、いくつかの損失関数を調べます。広範な実験により、私たちの方法は、合成データセットと実世界のデータセットの両方で既存の対になっていない排水アプローチに対して有利に機能し、完全に監視されたモデルまたは半監視されたモデルよりも優れていることが示されています。
Deep learning-based single image deraining (SID) with unpaired information is of immense importance, as relying on paired synthetic data often limits their generality and scalability in real-world applications. However, we noticed that direct employ of unpaired adversarial learning and cycle-consistency constraints in the SID task is insufficient to learn the underlying relationship from rainy input to clean outputs, since the domain knowledge between rainy and rain-free images is asymmetrical. To address such limitation, we develop an effective unpaired SID method which explores mutual properties of the unpaired exemplars by a contrastive learning manner in a GAN framework, named as CDR-GAN. The proposed method mainly consists of two cooperative branches: Bidirectional Translation Branch (BTB) and Contrastive Guidance Branch (CGB). Specifically, BTB takes full advantage of the circulatory architecture of adversarial consistency to exploit latent feature distributions and guide transfer ability between two domains by equipping it with bidirectional mapping. Simultaneously, CGB implicitly constrains the embeddings of different exemplars in rain space by encouraging the similar feature distributions closer while pushing the dissimilar further away, in order to better help rain removal and image restoration. During training, we explore several loss functions to further constrain the proposed CDR-GAN. Extensive experiments show that our method performs favorably against existing unpaired deraining approaches on both synthetic and real-world datasets, even outperforms several fully-supervised or semi-supervised models.
updated: Tue Sep 07 2021 10:00:45 GMT+0000 (UTC)
published: Tue Sep 07 2021 10:00:45 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アソシエイト