ImageSegmentationEM

Models

U-Net: Convolutional Networks for Biomedical Image Segmentation
- https://arxiv.org/abs/1505.04597
Recurrent Residual Convolutional Neural Network based on U-Net (R2U-Net) for Medical Image Segmentation
- https://arxiv.org/abs/1802.06955
Attention U-Net: Learning Where to Look for the Pancreas
- https://arxiv.org/abs/1804.03999
Attention R2U-Net : Just integration of two recent advanced works (R2U-Net + Attention U-Net)

python main.py -h

python main.py --model_type={} --epoch={} --result_path={} --lr={}

Thanks goes to these wonderful people (emoji key):

This project follows the all-contributors specification. Contributions of any kind welcome!