This repository including some extension implements based original VNet network.
- PyTorch>1.1.0
- inplace_abn
please see the installation in mapillary/inplace_abn
(PS: if you don't want the
inplace_abnmodule just comment line )
- Clone this repository
git clone https://github.com/Hsuxu/Magic-VNet.git
cd Magic-VNet
python setup.py install
- import to your code
from magic_vnet import VNet # network you need- V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
- Concurrent Spatial and Channel ‘Squeeze & Excitation’ in Fully Convolutional Networks
- Squeeze-and-Excitation Networks
- Pyramid Scene Parsing Network
- Attention U-Net: Learning Where to Look for the Pancreas
- Rethinking Atrous Convolution for Semantic Image Segmentation
- Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation
- Selective Kernel Networks
- Towards Stablizing Batch Statistics in Backward Propagation of Batch Normalization
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs