Medical Image segmentation

This project is about my personal experiments with various of Neural Network for medical image segmentation (organs, brain, tumors,...) implemented in PyTorch and PyTorch Lightning.

Dataset

• Dataset B
• LITS 2017

Neural Nets

• UNet
• AttentionUNet
• Nested-UNet
• STAN (Small tumor-aware network)
• MSDNet (Mixed-scale dense convolutional network)

Attention mechanism

• Attention Gated (paper)
• Grid Attention
• Axial Attention (paper)

Loss function

• Dice loss
• Lovasz loss
• Tversky loss

Demo