├── .gitignore ├── README.md ├── configs ├── isic │ ├── isic2018_attunet.yaml │ ├── isic2018_missformer.yaml │ ├── isic2018_multiresunet.yaml │ ├── isic2018_resunet.yaml │ ├── isic2018_transunet.yaml │ ├── isic2018_uctransnet.yaml │ ├── isic2018_unet.yaml │ └── isic2018_unetpp.yaml └── segpc │ ├── segpc2021_attunet.yaml │ ├── segpc2021_missformer.yaml │ ├── segpc2021_multiresunet.yaml │ ├── segpc2021_resunet.yaml │ ├── segpc2021_transunet.yaml │ ├── segpc2021_uctransnet.yaml │ ├── segpc2021_unet.yaml │ └── segpc2021_unetpp.yaml ├── datasets ├── README.md ├── __init__.py ├── isic.ipynb ├── isic.py ├── prepare_isic.ipynb ├── prepare_segpc.ipynb ├── segpc.ipynb └── segpc.py ├── images ├── ComparisonOfModels.png ├── U-Net_Taxonomy.png ├── isic2018.png ├── isic2018_sample.png ├── segpc.png ├── segpc2021_sample.png ├── synapse.png └── unet-pipeline.png ├── losses.py ├── models ├── __init__.py ├── _missformer │ ├── MISSFormer.py │ ├── __init__.py │ └── segformer.py ├── _resunet │ ├── __init__.py │ ├── modules.py │ └── res_unet.py ├── _transunet │ ├── vit_seg_configs.py │ ├── vit_seg_modeling.py │ ├── vit_seg_modeling_c4.py │ ├── vit_seg_modeling_resnet_skip.py │ └── vit_seg_modeling_resnet_skip_c4.py ├── _uctransnet │ ├── CTrans.py │ ├── Config.py │ ├── UCTransNet.py │ └── UNet.py ├── attunet.py ├── multiresunet.py ├── unet.py └── unetpp.py ├── train_and_test ├── isic │ ├── attunet-isic.ipynb │ ├── attunet-isic.py │ ├── missformer-isic.ipynb │ ├── missformer-isic.py │ ├── multiresunet-isic.ipynb │ ├── multiresunet-isic.py │ ├── resunet-isic.ipynb │ ├── resunet-isic.py │ ├── transunet-isic.ipynb │ ├── transunet-isic.py │ ├── uctransnet-isic.ipynb │ ├── uctransnet-isic.py │ ├── unet-isic.ipynb │ ├── unet-isic.py │ ├── unetpp-isic.ipynb │ └── unetpp-isic.py └── segpc │ ├── attunet-segpc.ipynb │ ├── attunet-segpc.py │ ├── missformer-segpc.ipynb │ ├── missformer-segpc.py │ ├── multiresunet-segpc.ipynb │ ├── multiresunet-segpc.py │ ├── resunet-segpc.ipynb │ ├── resunet-segpc.py │ ├── transunet-segpc.ipynb │ ├── transunet-segpc.py │ ├── uctransnet-segpc.ipynb │ ├── uctransnet-segpc.py │ ├── unet-segpc.ipynb │ ├── unet-segpc.py │ ├── unetpp-segpc.ipynb │ └── unetpp-segpc.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/README.md -------------------------------------------------------------------------------- /configs/isic/isic2018_attunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_attunet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_missformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_missformer.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_multiresunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_multiresunet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_resunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_resunet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_transunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_transunet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_uctransnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_uctransnet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_unet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_unet.yaml -------------------------------------------------------------------------------- /configs/isic/isic2018_unetpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/isic/isic2018_unetpp.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_attunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_attunet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_missformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_missformer.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_multiresunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_multiresunet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_resunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_resunet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_transunet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_transunet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_uctransnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_uctransnet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_unet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_unet.yaml -------------------------------------------------------------------------------- /configs/segpc/segpc2021_unetpp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/configs/segpc/segpc2021_unetpp.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/isic.ipynb -------------------------------------------------------------------------------- /datasets/isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/isic.py -------------------------------------------------------------------------------- /datasets/prepare_isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/prepare_isic.ipynb -------------------------------------------------------------------------------- /datasets/prepare_segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/prepare_segpc.ipynb -------------------------------------------------------------------------------- /datasets/segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/segpc.ipynb -------------------------------------------------------------------------------- /datasets/segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/datasets/segpc.py -------------------------------------------------------------------------------- /images/ComparisonOfModels.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/ComparisonOfModels.png -------------------------------------------------------------------------------- /images/U-Net_Taxonomy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/U-Net_Taxonomy.png -------------------------------------------------------------------------------- /images/isic2018.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/isic2018.png -------------------------------------------------------------------------------- /images/isic2018_sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/isic2018_sample.png -------------------------------------------------------------------------------- /images/segpc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/segpc.png -------------------------------------------------------------------------------- /images/segpc2021_sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/segpc2021_sample.png -------------------------------------------------------------------------------- /images/synapse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/synapse.png -------------------------------------------------------------------------------- /images/unet-pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/images/unet-pipeline.png -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/losses.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/_missformer/MISSFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_missformer/MISSFormer.py -------------------------------------------------------------------------------- /models/_missformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/_missformer/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_missformer/segformer.py -------------------------------------------------------------------------------- /models/_resunet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/_resunet/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_resunet/modules.py -------------------------------------------------------------------------------- /models/_resunet/res_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_resunet/res_unet.py -------------------------------------------------------------------------------- /models/_transunet/vit_seg_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_transunet/vit_seg_configs.py -------------------------------------------------------------------------------- /models/_transunet/vit_seg_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_transunet/vit_seg_modeling.py -------------------------------------------------------------------------------- /models/_transunet/vit_seg_modeling_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_transunet/vit_seg_modeling_c4.py -------------------------------------------------------------------------------- /models/_transunet/vit_seg_modeling_resnet_skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_transunet/vit_seg_modeling_resnet_skip.py -------------------------------------------------------------------------------- /models/_transunet/vit_seg_modeling_resnet_skip_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_transunet/vit_seg_modeling_resnet_skip_c4.py -------------------------------------------------------------------------------- /models/_uctransnet/CTrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_uctransnet/CTrans.py -------------------------------------------------------------------------------- /models/_uctransnet/Config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_uctransnet/Config.py -------------------------------------------------------------------------------- /models/_uctransnet/UCTransNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_uctransnet/UCTransNet.py -------------------------------------------------------------------------------- /models/_uctransnet/UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/_uctransnet/UNet.py -------------------------------------------------------------------------------- /models/attunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/attunet.py -------------------------------------------------------------------------------- /models/multiresunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/multiresunet.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/unet.py -------------------------------------------------------------------------------- /models/unetpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/models/unetpp.py -------------------------------------------------------------------------------- /train_and_test/isic/attunet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/attunet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/attunet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/attunet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/missformer-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/missformer-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/missformer-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/missformer-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/multiresunet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/multiresunet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/multiresunet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/multiresunet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/resunet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/resunet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/resunet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/resunet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/transunet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/transunet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/transunet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/transunet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/uctransnet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/uctransnet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/uctransnet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/uctransnet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/unet-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/unet-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/unet-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/unet-isic.py -------------------------------------------------------------------------------- /train_and_test/isic/unetpp-isic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/unetpp-isic.ipynb -------------------------------------------------------------------------------- /train_and_test/isic/unetpp-isic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/isic/unetpp-isic.py -------------------------------------------------------------------------------- /train_and_test/segpc/attunet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/attunet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/attunet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/attunet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/missformer-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/missformer-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/missformer-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/missformer-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/multiresunet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/multiresunet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/multiresunet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/multiresunet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/resunet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/resunet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/resunet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/resunet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/transunet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/transunet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/transunet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/transunet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/uctransnet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/uctransnet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/uctransnet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/uctransnet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/unet-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/unet-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/unet-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/unet-segpc.py -------------------------------------------------------------------------------- /train_and_test/segpc/unetpp-segpc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/unetpp-segpc.ipynb -------------------------------------------------------------------------------- /train_and_test/segpc/unetpp-segpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/train_and_test/segpc/unetpp-segpc.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NITR098/Awesome-U-Net/HEAD/utils.py --------------------------------------------------------------------------------