DeepAI AI Chat
Log In Sign Up

WaveSNet: Wavelet Integrated Deep Networks for Image Segmentation

05/29/2020
by   Qiufu Li, et al.
Shenzhen University
20

In deep networks, the lost data details significantly degrade the performances of image segmentation. In this paper, we propose to apply Discrete Wavelet Transform (DWT) to extract the data details during feature map down-sampling, and adopt Inverse DWT (IDWT) with the extracted details during the up-sampling to recover the details. We firstly transform DWT/IDWT as general network layers, which are applicable to 1D/2D/3D data and various wavelets like Haar, Cohen, and Daubechies, etc. Then, we design wavelet integrated deep networks for image segmentation (WaveSNets) based on various architectures, including U-Net, SegNet, and DeepLabv3+. Due to the effectiveness of the DWT/IDWT in processing data details, experimental results on CamVid, Pascal VOC, and Cityscapes show that our WaveSNets achieve better segmentation performances than their vanilla versions.

READ FULL TEXT

page 1

page 5

page 6

05/07/2020

Wavelet Integrated CNNs for Noise-Robust Image Classification

Convolutional Neural Networks (CNNs) are generally prone to noise interr...
01/08/2020

DC-WCNN: A deep cascade of wavelet based convolutional neural networks for MR Image Reconstruction

Several variants of Convolutional Neural Networks (CNN) have been develo...
07/28/2021

WaveCNet: Wavelet Integrated CNNs to Suppress Aliasing Effect for Noise-Robust Image Classification

Though widely used in image classification, convolutional neural network...
06/01/2021

3D WaveUNet: 3D Wavelet Integrated Encoder-Decoder Network for Neuron Segmentation

3D neuron segmentation is a key step for the neuron digital reconstructi...
04/30/2018

Stack-U-Net: Refinement Network for Image Segmentation on the Example of Optic Disc and Cup

In this work, we propose a special cascade network for image segmentatio...
07/14/2019

A Divide-and-Conquer Approach towards Understanding Deep Networks

Deep neural networks have achieved tremendous success in various fields ...

Code Repositories