A residual network framework based on weighted feature channels for multispectral image compression

Abstract

Deep learning has achieved great success in many computer vision tasks, such as recognition, image enhancement and image compression. However, it is difficult to use a residual network that can efficiently consider the characteristics of multispectral images for multispectral image compression. In this paper, a novel end-to-end multispectral image compression framework based on a weighted feature channel residual network is proposed to efficiently remove the spatial and spectral redundancy of multispectral images by extracting the importance of each channel. The multispectral image compression framework includes a forward coding network, a rate-distortion optimizer, a quantizer/inverse quantizer, an entropy encoder/decoder and an inverse decoding network. In the encoder, multispectral images are directly fed into the forward coding network, and the main spectral and spatial features of the multispectral images are extracted by the residual block. Additionally, the weighted feature channel module can explicitly model the relationship between feature channels when extracting features from multispectral images and adaptively allocate different weights for each feature channel through training. The rate-distortion optimizer is added to make the main features compact. Then, the intermediate feature data are quantized and encoded by lossless entropy coding to obtain a code stream. In the decoder, the code stream is approximately restored to the intermediate features through the entropy decoder and the inverse quantizer. Then, the intermediate features are reconstructed to the multispectral images by the inverse decoding network. Experimental results on 7-band multispectral images of the Landsat 8 satellite and 8-band multispectral images of WorldView-3 satellite demonstrate that the proposed algorithm can achieve a better PSNR than conventional 2D schemes (which are JPEG2000 and JPEG in this paper) and 3D scheme (which is 3D-SPIHT in this paper) and can effectively preserve more spectral information of multispectral images.