Aiming at the problem that the existing defogging algorithm is not ideal for defogging
a single image defogging algorithm based on attention mechanism is proposed. Channel attention and pixel attention are constructed by introducing attention mechanism
and the feature attention module is realized by combining the two. Then the basic module is built through multi-scale convolution
local residual learning
and feature attention. At last
the end-to-end defogging is realized by global residual learning. The experimental results show that the algorithm is superior to the comparison algorithm in three image evaluation indexes of peak signal-to-noise ratio(PSNR)
structural similarity(SSIM) and feature similarity(FSIM)
and it achieves a good defogging effect and effectively solves the problems of undesirable defogging effect
关键词
Keywords
references
LI B Y, PENG X L, WANG Z Y, et al. AOD-Net: all-in-one dehazing network[C]//Proceedings of the 2017 IEEE International Conference on Computer Vision (ICCV). Venice: IEEE, 2017: 4770-4778. DOI: 10.1109/ICCV.2017.511.
HE K M, SUN J, TANG X O. Single image haze removal using dark channel prior[J]. IEEE Transactions on PatternAnalysis andMachine Intelligence, 2011, 33(12): 2341-2353. DOI:10.1109/TPAMI.2010.168.
LI B Y, REN W Q, FU D P, et al. Benchmarking single-image dehazing and beyond[J]. IEEE Transactions on Image Processing, 2019, 28(1): 492-505. DOI: 10.1109/TIP.2018.2867951.
CAI B L, XU X M, JIA K, et al. DehazeNet: an end-to-end system for single image haze removal[J]. IEEE Transactions on Image Processing, 2016, 25(11): 5187-5198. DOI: 10.1109/TIP.2016.2598681.
LIU X H, MA Y R, SHI Z H, et al. GridDehazeNet: attention-based multi-scale network for image dehazing[C]//Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision (ICCV). Seoul: IEEE, 2019: 7314-7323. DOI: 10.1109/ICCV.2019.00741.
BERMAN D, TREIBITZ T, AVIDAN S. Non-local imagedehazing[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas: IEEE, 2016: 1674-1682. DOI: 10.1109/CVPR.2016.185.
CHEN D D, HE M M, FAN Q N, et al. Gated context aggregation network for image dehazing and deraining[C]//Proceedings of the2019IEEE Winter Conference on Applications of Computer Vision (WACV). Waikoloa Village: IEEE, 2019: 1375-1383. DOI: 10.1109/WACV.2019.00151.