VNet PyTorch Implements

This repository including some extension implements based original VNet network.

Requirements

1. PyTorch>1.1.0
2. inplace_abn please see the installation in mapillary/inplace_abn (PS: if you don't want the inplace_abn module just comment line )

Installation & Usage

• 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

Reference

Paper

• 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

Code

• ai-med/squeeze_and_excitation
• mapillary/inplace_abn
• megvii-model/MABN
• pppLang/SKNet