xzwthu / SegUNet

U-Net structure compatible with 2D data and 3D data, and different loss functions are implemented.

U-Net structure compatible with 2D data and 3D data, and different loss functions are implemented.

Currently supported loss functions

• focal loss
• dice loss
• cross entropy loss
• distance loss
• edge loss

Currently supported network structures

• U-Net
• V-Net
• U-Net++
• attention U-Net
• deeplab v1-3

support multi-GPU training, synchronized batch normalization