arXiv reaDer
対照学習と実世界のデータセットによる水中画像の復元
Underwater Image Restoration via Contrastive Learning and a Real-world Dataset
水中の画像の復元は、水中の世界を明らかにする上で非常に重要です。過去数十年の間に、数多くの技術とアルゴリズムが開発されてきました。ただし、イメージング/センシング、照明、および屈折の幾何学的歪みに関連する基本的な問題のため、鮮明な水中画像をキャプチャする際に、水中画像の復元に関する包括的な評価は行われていません。このギャップに対処するために、既存の方法のベンチマークと新しい深層学習ベースの方法の開発をサポートする目的で、「HICRD」(ヘロン島サンゴ礁データセット)と呼ばれる大規模な実際の水中画像データセットを構築しました。参照画像の生成には、正確な水パラメータ(拡散減衰係数)を採用しています。対になっていないトレーニングセットには、2000個の参照復元画像と6003個の元の水中画像があります。さらに、教師なし画像から画像への変換フレームワークに基づく水中画像復元の新しい方法を提示します。私たちの提案する方法は、対照的な学習と生成的敵対的ネットワークを活用して、生の画像と復元された画像の間の相互情報量を最大化しました。最近のアプローチと比較した広範な実験は、提案された方法の優位性をさらに示しています。コードとデータセットはGitHubで公開されています。
Underwater image restoration is of significant importance in unveiling the underwater world. Numerous techniques and algorithms have been developed in the past decades. However, due to fundamental difficulties associated with imaging/sensing, lighting, and refractive geometric distortions, in capturing clear underwater images, no comprehensive evaluations have been conducted of underwater image restoration. To address this gap, we have constructed a large-scale real underwater image dataset, dubbed `HICRD' (Heron Island Coral Reef Dataset), for the purpose of benchmarking existing methods and supporting the development of new deep-learning based methods. We employ accurate water parameter (diffuse attenuation coefficient) in generating reference images. There are 2000 reference restored images and 6003 original underwater images in the unpaired training set. Further, we present a novel method for underwater image restoration based on unsupervised image-to-image translation framework. Our proposed method leveraged contrastive learning and generative adversarial networks to maximize the mutual information between raw and restored images. Extensive experiments with comparisons to recent approaches further demonstrate the superiority of our proposed method. Our code and dataset are publicly available at GitHub.
updated: Sun Jun 20 2021 16:06:26 GMT+0000 (UTC)
published: Sun Jun 20 2021 16:06:26 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アソシエイト