Home

Awesome

Multi-level Wavelet-CNN for Image Restoration

Abstract

The tradeoff between receptive field size and efficiency is a crucial issue in low level vision. Plain convolutional networks (CNNs) generally enlarge the receptive field at the expense of computational cost. Recently, dilated filtering has been adopted to address this issue. But it suffers from gridding effect, and the resulting receptive field is only a sparse sampling of input image with checkerboard patterns. In this paper, we present a novel multi-level wavelet CNN (MWCNN) model for better tradeoff between receptive field size and computational efficiency. With the modified U-Net architecture, wavelet transform is introduced to reduce the size of feature maps in the contracting subnetwork. Furthermore, another convolutional layer is further used to decrease the channels of feature maps. In the expanding subnetwork, inverse wavelet transform is then deployed to reconstruct the high resolution feature maps. Our MWCNN can also be explained as the generalization of dilated filtering and subsampling, and can be applied to many image restoration tasks. The experimental results clearly show the effectiveness of MWCNN for image denoising, single image super-resolution, and JPEG image artifacts removal.

Network Architecture

Multi-level wavelet-CNN architecture. It consists two parts: the contracting and expanding subnetworks. Each solid box corresponds to a multi-channel feature map. And the number of channels is annotated on the top of the box. The network depth is 24. Moreover, our MWCNN can be further extended to higher level (e.g., ≥ 4) by duplicating the configuration of the 3rd level subnetwork.

<p align="center"> <img src="./images/Architecture.png" width="100%"> <br /> <small> </small> </p>

Results

Image denoising

<p align="center"> <img src="./images/GD_RESL.png" width="100%"> <br /> <small> </small> </p> <p align="center"> <img src="./images/GD_pic.png" width="100%"> <br /> <small> </small> </p>

SISR

<p align="center"> <img src="./images/SISR_RESL.png" width="100%"> <br /> <small> </small> </p> <p align="center"> <img src="./images/SISR_pic.png" width="100%"> <br /> <small> </small> </p>

JPEG Artifacts Removal

<p align="center"> <img src="./images/JAR_RESL.png" width="100%"> <br /> <small> </small> </p> <p align="center"> <img src="./images/JAR_pic.png" width="100%"> <br /> <small> </small> </p>

Test models

Download the pre-trained model with the following url and put it into ./models

Image denoising

'Demo_Image_Denoising.m' is the demo of MWCNN for image denoising.

SISR

'Demo_SISR.m' is the demo of MWCNN for SISR.

JPEG Image Artifacts Removal

'Demo_JPEG_Image_ Artifacts_Removal.m' is the demo of MWCNN for JPEG image artifacts removal.

Requirements and Dependencies

Contact

Please send email to lpj008@126.com

Citation

@InProceedings{Liu_2018_CVPR_Workshops,
author = {Liu, Pengju and Zhang, Hongzhi and Zhang, Kai and Lin, Liang and Zuo, Wangmeng},
title = {Multi-Level Wavelet-CNN for Image Restoration},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {June},
year = {2018}
}