arXiv reaDer
大雨の除去から細部の復元まで:より速くより良いネットワーク
From Heavy Rain Removal to Detail Restoration: A Faster and Better Network
大雨による濃い雨の蓄積は、画像を大幅に洗い流し、画像の背景の詳細を破壊する可能性があります。既存の大雨除去モデルは大雨除去のパフォーマンスを向上させますが、それらのほとんどは雨のない画像の詳細な再構成の精度を無視していることがわかります。本論文では、構造精度の雨のない画像で効果的な排水を実現するために、デュアルステージプログレッシブエンハンスメントネットワーク(DPENet)を提案します。フレームワークには、レインストリーク除去ネットワーク(R ^ 2Net)と詳細再構築ネットワーク(DRNet)の2つの主要モジュールが含まれています。前者は正確な雨の除去を達成することを目的とし、後者は雨のない画像の詳細を復元するように設計されています。雨が降らない画像の詳細な復元と排水の効果の間のトレードオフを達成するために、ネットワーク内に2つの主要な戦略を紹介します。最初に、大雨の高/低レベルの特徴を集約するために、レインストリーク除去ネットワーク内の拡張された高密度残留ブロック(DDRB)が提示されます。第二に、詳細再構成ネットワーク内の強化された残留ピクセルごとの注意ブロック(ERPAB)は、コンテキスト情報の集約のために設計されています。また、雨のない画像の限界および地域の精度を強調するための包括的な損失関数を提案します。ベンチマーク公開データセットに関する広範な実験は、大雨を除去するための構造を維持する雨のない画像を実現する上で、提案された方法の効率と有効性の両方を示しています。ソースコードと事前にトレーニングされたモデルは、https://github.com/wybchd/DPENetにあります。
The dense rain accumulation in heavy rain can significantly wash out images and thus destroy the background details of images. Although existing deep rain removal models lead to improved performance for heavy rain removal, we find that most of them ignore the detail reconstruction accuracy of rain-free images. In this paper, we propose a dual-stage progressive enhancement network (DPENet) to achieve effective deraining with structure-accurate rain-free images. Two main modules are included in our framework, namely a rain streaks removal network (R^2Net) and a detail reconstruction network (DRNet). The former aims to achieve accurate rain removal, and the latter is designed to recover the details of rain-free images. We introduce two main strategies within our networks to achieve trade-off between the effectiveness of deraining and the detail restoration of rain-free images. Firstly, a dilated dense residual block (DDRB) within the rain streaks removal network is presented to aggregate high/low level features of heavy rain. Secondly, an enhanced residual pixel-wise attention block (ERPAB) within the detail reconstruction network is designed for context information aggregation. We also propose a comprehensive loss function to highlight the marginal and regional accuracy of rain-free images. Extensive experiments on benchmark public datasets show both efficiency and effectiveness of the proposed method in achieving structure-preserving rain-free images for heavy rain removal. The source code and pre-trained models can be found at https://github.com/wybchd/DPENet.
updated: Sat May 07 2022 04:55:05 GMT+0000 (UTC)
published: Sat May 07 2022 04:55:05 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アソシエイト