この論文では、単一画像の曇り除去スキームのためのガイド付き伝送マップ(EDN-GTM)を備えた新しいエンコーダ-デコーダネットワークを提案します。提案されたEDN-GTMは、ネットワークの入力として事前にダークチャネルを採用することによって推定された伝送マップと組み合わせて、従来のRGBヘイジー画像を取得します。提案されたEDN-GTMは、コアネットワークとして画像セグメンテーションにU-Netを利用し、空間ピラミッドプーリングモジュールやSwishアクティベーションなどのさまざまな変更を利用して、最先端の曇り除去性能を実現します。ベンチマークデータセットでの実験は、提案されたEDN-GTMが、PSNRおよびSSIMメトリックの点で、従来のディープラーニングベースの画像デヘイズスキームのほとんどを上回っていることを示しています。提案されたEDN-GTMは、オブジェクト検出問題への適用性をさらに証明します。具体的には、物体検出を駆動するための画像前処理ツールに適用すると、提案されたEDN-GTMは、ヘイズを効率的に除去し、mAP測定に関して検出精度を4.73%大幅に向上させることができます。コードはhttps://github.com/tranleanh/edn-gtmで入手できます。
A novel Encoder-Decoder Network with Guided Transmission Map (EDN-GTM) for single image dehazing scheme is proposed in this paper. The proposed EDN-GTM takes conventional RGB hazy image in conjunction with its transmission map estimated by adopting dark channel prior as the inputs of the network. The proposed EDN-GTM utilizes U-Net for image segmentation as the core network and utilizes various modifications including spatial pyramid pooling module and Swish activation to achieve state-of-the-art dehazing performance. Experiments on benchmark datasets show that the proposed EDN-GTM outperforms most of traditional and deep learning-based image dehazing schemes in terms of PSNR and SSIM metrics. The proposed EDN-GTM furthermore proves its applicability to object detection problems. Specifically, when applied to an image preprocessing tool for driving object detection, the proposed EDN-GTM can efficiently remove haze and significantly improve detection accuracy by 4.73% in terms of mAP measure. The code is available at: https://github.com/tranleanh/edn-gtm.