arXiv reaDer
単一画像の曇り除去用のビジョントランスフォーマー
Vision Transformers for Single Image Dehazing
画像の曇り除去は、かすんでいる画像から潜在的なかすみのない画像を推定する代表的な低レベルの視覚タスクです。近年、畳み込みニューラルネットワークベースの方法が画像の曇り除去を支配してきました。ただし、最近高レベルの視覚タスクで画期的な進歩を遂げたビジョントランスフォーマーは、画像の曇り除去に新しい次元をもたらしていません。人気のあるSwinTransformerから始めて、その主要なデザインのいくつかが画像の曇り除去に適していないことを発見しました。この目的のために、修正された正規化層、活性化関数、空間情報集約スキームなどのさまざまな改善で構成されるDehazeFormerを提案します。 DehazeFormerの複数のバリアントをさまざまなデータセットでトレーニングして、その有効性を示します。具体的には、最も頻繁に使用されるSOTS屋内セットでは、小型モデルがFFA-Netを上回り、#Paramはわずか25%、計算コストは5%です。私たちの知る限り、私たちの大型モデルは、SOTS屋内セットでPSNRが40 dBを超える最初の方法であり、以前の最先端の方法を劇的に上回っています。また、非常に不均一なヘイズを除去するメソッドの機能を評価するために、大規模で現実的なリモートセンシングのヘイズ除去データセットを収集します。
Image dehazing is a representative low-level vision task that estimates latent haze-free images from hazy images. In recent years, convolutional neural network-based methods have dominated image dehazing. However, vision Transformers, which has recently made a breakthrough in high-level vision tasks, has not brought new dimensions to image dehazing. We start with the popular Swin Transformer and find that several of its key designs are unsuitable for image dehazing. To this end, we propose DehazeFormer, which consists of various improvements, such as the modified normalization layer, activation function, and spatial information aggregation scheme. We train multiple variants of DehazeFormer on various datasets to demonstrate its effectiveness. Specifically, on the most frequently used SOTS indoor set, our small model outperforms FFA-Net with only 25% #Param and 5% computational cost. To the best of our knowledge, our large model is the first method with the PSNR over 40 dB on the SOTS indoor set, dramatically outperforming the previous state-of-the-art methods. We also collect a large-scale realistic remote sensing dehazing dataset for evaluating the method's capability to remove highly non-homogeneous haze.
updated: Fri Apr 08 2022 07:17:20 GMT+0000 (UTC)
published: Fri Apr 08 2022 07:17:20 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アソシエイト