arXiv reaDer
GridFormer: 悪天候時の画像復元のためのグリッド構造を備えた残留高密度変圧器
GridFormer: Residual Dense Transformer with Grid Structure for Image Restoration in Adverse Weather Conditions
コンピュータービジョンにおいて、悪天候下での画像復元は困難な作業です。この論文では、悪天候条件下での画像復元のバックボーンとして機能する GridFormer と呼ばれる新しいトランスフォーマーベースのフレームワークを提案します。 GridFormer は、残留高密度トランス ブロックを使用したグリッド構造で設計されており、2 つのコア設計が導入されています。まず、トランス層で強化されたアテンション メカニズムを使用します。このメカニズムには、効率を向上させるためのサンプラーとコンパクトセルフアテンションのステージ、およびローカル情報を強化するためのローカルエンハンスメントステージが含まれています。次に、最後の GridFormer 層として残留密変換ブロック (RDTB) を導入します。この設計により、以前のローカル機能と現在のローカル機能の両方から有効な機能を学習するネットワークの能力がさらに向上します。 GridFormer フレームワークは、画像のディレイニング、かすみ除去、ディレイニングとかすみの除去、除雪、および複数の天候の復元を含む、悪天候条件下での 5 つの多様な画像復元タスクで最先端の結果を達成します。ソースコードと事前トレーニング済みモデルがリリースされます。
Image restoration in adverse weather conditions is a difficult task in computer vision. In this paper, we propose a novel transformer-based framework called GridFormer which serves as a backbone for image restoration under adverse weather conditions. GridFormer is designed in a grid structure using a residual dense transformer block, and it introduces two core designs. First, it uses an enhanced attention mechanism in the transformer layer. The mechanism includes stages of the sampler and compact self-attention to improve efficiency, and a local enhancement stage to strengthen local information. Second, we introduce a residual dense transformer block (RDTB) as the final GridFormer layer. This design further improves the network's ability to learn effective features from both preceding and current local features. The GridFormer framework achieves state-of-the-art results on five diverse image restoration tasks in adverse weather conditions, including image deraining, dehazing, deraining & dehazing, desnowing, and multi-weather restoration. The source code and pre-trained models will be released.
updated: Mon May 29 2023 03:03:53 GMT+0000 (UTC)
published: Mon May 29 2023 03:03:53 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アソシエイト