arXiv reaDer
SDNet: swin を使用した単一イメージの排出のための mutil-branch
SDNet: mutil-branch for single image deraining using swin
雨のすじは画質を低下させ、自動運転や社会保障などの後続のコンピュータ ビジョン タスクのパフォーマンスに深刻な影響を及ぼします。したがって、特定の雨の画像から雨のすじを取り除くことは非常に重要です。畳み込みニューラル ネットワーク (CNN) は、画像の枯渇タスクで広く使用されていますが、畳み込み演算の局所的な計算特性により、画像の枯渇タスクの開発が制限されます。最近、人気のあるトランスフォーマーには、画像を枯渇させるタスクの開発をさらに促進できるグローバルな計算機能があります。この論文では、Swin-transformer を画像流出の分野に初めて導入して、画像流出の分野における Swin-transformer の性能と可能性を研究します。具体的には、Swin-transformer の基本モジュールを改善し、単一画像の雨の除去を実装するための 3 分岐モデルを設計します。前者は基本的な雨模様の特徴抽出を実装し、後者はさまざまな特徴を融合して画像の特徴をさらに抽出および処理します。また、ジャンプ接続を採用し、深い特徴と浅い特徴を融合。実験に関して言えば、既存の公開データセットは画像の重複と比較的均質な背景の影響を受けます。そこで、モデルを検証するための新しいデータセット Rain3000 を提案します。したがって、モデルを検証するための新しいデータセット Rain3000 を提案します。公開されているデータセットRain100L、Rain100H、およびデータセットRain3000の実験結果は、提案された方法が、現在の主流の単一画像の雨ストリーク除去モデルよりもパフォーマンスと推論速度の利点があることを示しています.ソースコードはhttps://githubで入手できます. com/H-tfx/SDNet.
Rain streaks degrade the image quality and seriously affect the performance of subsequent computer vision tasks, such as autonomous driving, social security, etc. Therefore, removing rain streaks from a given rainy images is of great significance. Convolutional neural networks(CNN) have been widely used in image deraining tasks, however, the local computational characteristics of convolutional operations limit the development of image deraining tasks. Recently, the popular transformer has global computational features that can further facilitate the development of image deraining tasks. In this paper, we introduce Swin-transformer into the field of image deraining for the first time to study the performance and potential of Swin-transformer in the field of image deraining. Specifically, we improve the basic module of Swin-transformer and design a three-branch model to implement single-image rain removal. The former implements the basic rain pattern feature extraction, while the latter fuses different features to further extract and process the image features. In addition, we employ a jump connection to fuse deep features and shallow features. In terms of experiments, the existing public dataset suffers from image duplication and relatively homogeneous background. So we propose a new dataset Rain3000 to validate our model. Therefore, we propose a new dataset Rain3000 for validating our model. Experimental results on the publicly available datasets Rain100L, Rain100H and our dataset Rain3000 show that our proposed method has performance and inference speed advantages over the current mainstream single-image rain streaks removal models.The source code will be available at https://github.com/H-tfx/SDNet.
updated: Mon May 31 2021 16:06:02 GMT+0000 (UTC)
published: Mon May 31 2021 16:06:02 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アソシエイト