arXiv reaDer
二重対照学習を使用した対になっていない深い画像のドレイン
Unpaired Deep Image Deraining Using Dual Contrastive Learning
ペアになっている実世界のデータを取得することはほとんど不可能であるため、不対のクリーンで雨の多い画像のセットから単一画像ドレイン(SID)ネットワークを学習することは実用的で価値があります。ただし、教師としてペアのデータがないと、SIDネットワークの学習は困難です。さらに、SIDタスクで既存の対になっていない学習方法(たとえば、対になっていない敵対的学習やサイクル整合性の制約)を使用するだけでは、雨の画像ときれいな画像の間に大きなドメインギャップが存在するため、雨の入力からきれいな出力までの基本的な関係を学習するには不十分です。この論文では、DCD-GANと呼ばれる深い特徴空間での二重対照学習法によって対になっていない模範の相互特性を調査する効果的な対になっていないSIDの敵対的フレームワークを開発します。提案された方法は、主に双方向翻訳ブランチ(BTB)と対照ガイダンスブランチ(CGB)の2つの協調ブランチで構成されています。具体的には、BTBは、敵対的一貫性の循環アーキテクチャの利点を最大限に活用して、豊富な模範ペアを生成し、双方向マッピングを装備することで2つのドメイン間の潜在的な特徴分布を掘削します。同時に、CGBは、雨の除去をより容易にし、画像の復元を支援するために、類似した特徴分布をより近くに促し、異なるものをさらに遠ざけることによって、深い特徴空間への異なるエグザンプラの埋め込みを暗黙的に制限します。広範な実験は、私たちの方法が、合成データセットと実世界のデータセットの両方で既存の対になっていない排水アプローチに対して有利に機能し、いくつかの完全教師ありモデルまたは半教師ありモデルに対して同等の結果を生成することを示しています。
Learning single image deraining (SID) networks from an unpaired set of clean and rainy images is practical and valuable as acquiring paired real-world data is almost infeasible. However, without the paired data as the supervision, learning a SID network is challenging. Moreover, simply using existing unpaired learning methods (e.g., unpaired adversarial learning and cycle-consistency constraints) in the SID task is insufficient to learn the underlying relationship from rainy inputs to clean outputs as there exists significant domain gap between the rainy and clean images. In this paper, we develop an effective unpaired SID adversarial framework which explores mutual properties of the unpaired exemplars by a dual contrastive learning manner in a deep feature space, named as DCD-GAN. The proposed method mainly consists of two cooperative branches: Bidirectional Translation Branch (BTB) and Contrastive Guidance Branch (CGB). Specifically, BTB exploits full advantage of the circulatory architecture of adversarial consistency to generate abundant exemplar pairs and excavates latent feature distributions between two domains by equipping it with bidirectional mapping. Simultaneously, CGB implicitly constrains the embeddings of different exemplars in the deep feature space by encouraging the similar feature distributions closer while pushing the dissimilar further away, in order to better facilitate rain removal and help image restoration. Extensive experiments demonstrate that our method performs favorably against existing unpaired deraining approaches on both synthetic and real-world datasets, and generates comparable results against several fully-supervised or semi-supervised models.
updated: Mon Mar 14 2022 05:22:19 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アソシエイト