An All-in-One Network for Dehazing and Beyond
About
This paper proposes an image dehazing model built with a convolutional neural network (CNN), called All-in-One Dehazing Network (AOD-Net). It is designed based on a re-formulated atmospheric scattering model. Instead of estimating the transmission matrix and the atmospheric light separately as most previous models did, AOD-Net directly generates the clean image through a light-weight CNN. Such a novel end-to-end design makes it easy to embed AOD-Net into other deep models, e.g., Faster R-CNN, for improving high-level task performance on hazy images. Experimental results on both synthesized and natural hazy image datasets demonstrate our superior performance than the state-of-the-art in terms of PSNR, SSIM and the subjective visual quality. Furthermore, when concatenating AOD-Net with Faster R-CNN and training the joint pipeline from end to end, we witness a large improvement of the object detection performance on hazy images.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Dehazing | SOTS (test) | PSNR19.06 | 161 | |
| Image Dehazing | SOTS Outdoor | PSNR24.08 | 112 | |
| Image Dehazing | SOTS Indoor | PSNR21.01 | 62 | |
| Image Dehazing | Haze4k (test) | PSNR17.15 | 57 | |
| Image Dehazing | HazeRD (test) | PSNR15.63 | 15 | |
| Image Dehazing | Synthetic B (test) | SSIM83.25 | 7 | |
| Image Dehazing | A (test) | SSIM0.8842 | 6 |