An Improved U-Net Architecture for Image Dehazing
- 1 December 2021
- journal article
- research article
- Published by Institute of Electronics, Information and Communications Engineers (IEICE) in IEICE Transactions on Information and Systems
- Vol. E104.D (12), 2218-2225
- https://doi.org/10.1587/transinf.2021edp7043
Abstract
In this paper, we present a simple yet powerful deep neural network for natural image dehazing. The proposed method is designed based on U-Net architecture and we made some design changes to make it better. We first use Group Normalization to replace Batch Normalization to solve the problem of insufficient batch size due to hardware limitations. Second, we introduce FReLU activation into the U-Net block, which can achieve capturing complicated visual layouts with regular convolutions. Experimental results on public benchmarks demonstrate the effectiveness of the modified components. On the SOTS Indoor and Outdoor datasets, it obtains PSNR of 32.23 and 31.64 respectively, which are comparable performances with state-of-the-art methods. The code is publicly available online soon.Keywords
This publication has 25 references indexed in Scilit:
- Single Image Dehazing via Multi-scale Convolutional Neural NetworksPublished by Springer Science and Business Media LLC ,2016
- Perceptual Losses for Real-Time Style Transfer and Super-ResolutionPublished by Springer Science and Business Media LLC ,2016
- DehazeNet: An End-to-End System for Single Image Haze RemovalIEEE Transactions on Image Processing, 2016
- Learning Depth from Single Monocular Images Using Deep Convolutional Neural FieldsIEEE Transactions on Pattern Analysis and Machine Intelligence, 2015
- Fast R-CNNPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2015
- U-Net: Convolutional Networks for Biomedical Image SegmentationPublished by Springer Science and Business Media LLC ,2015
- A Fast Single Image Haze Removal Algorithm Using Color Attenuation PriorIEEE Transactions on Image Processing, 2015
- ImageNet Large Scale Visual Recognition ChallengeInternational Journal of Computer Vision, 2015
- Indoor Segmentation and Support Inference from RGBD ImagesLecture Notes in Computer Science, 2012
- Single image dehazingACM Transactions on Graphics, 2008