近年、深層学習に基づいた方法は、雨の除去において大きな進歩を遂げました。ただし、既存の方法は通常、良好な一般化能力を持たないため、ほとんどすべての既存の方法は特定の種類の雨縞を除去するのに十分な性能を持っていますが、他の種類の雨縞の性能は比較的低い可能性があります。本論文では、単一の画像から複数のタイプの雨縞を除去することを目的として、より優れた一般化能力を持つ新しい排水フレームワーク(GRASPP-GAN)を提案します。具体的には、雨の画像の深い特徴を抽出する修正されたResNet-18と、さまざまな形状とサイズの雨の縞に適応する修正されたASPP構造が一緒に構成されて、排水ネットワークのバックボーンを形成します。勾配領域での雨の縞のより顕著な特性を考慮して、勾配損失を導入して、トレーニングプロセスの監視を支援します。そのため、勾配情報を柔軟に抽出するためにSobel畳み込み層が構築されます。パフォーマンスをさらに向上させるために、提案されたネットワークを訓練するために敵対学習スキームが初めて採用されました。実際のデータセットと合成データセットの両方に関する広範な実験により、本方法が最先端の排水方法よりも定量的および定性的に優れていることが実証されています。さらに、変更を加えることなく、提案されたフレームワークは、かすみ除去時に優れた視覚パフォーマンスを実現します。
In recent years, deep learning based methods have made significant progress in rain-removing. However, the existing methods usually do not have good generalization ability, which leads to the fact that almost all of existing methods have a satisfied performance on removing a specific type of rain streaks, but may have a relatively poor performance on other types of rain streaks. In this paper, aiming at removing multiple types of rain streaks from single images, we propose a novel deraining framework (GRASPP-GAN), which has better generalization capacity. Specifically, a modified ResNet-18 which extracts the deep features of rainy images and a revised ASPP structure which adapts to the various shapes and sizes of rain streaks are composed together to form the backbone of our deraining network. Taking the more prominent characteristics of rain streaks in the gradient domain into consideration, a gradient loss is introduced to help to supervise our deraining training process, for which, a Sobel convolution layer is built to extract the gradient information flexibly. To further boost the performance, an adversarial learning scheme is employed for the first time to train the proposed network. Extensive experiments on both real-world and synthetic datasets demonstrate that our method outperforms the state-of-the-art deraining methods quantitatively and qualitatively. In addition, without any modifications, our proposed framework also achieves good visual performance on dehazing.