├── 2D_Dense_UNet.py ├── 2D_FPN.py ├── 2D_Linknet.py ├── 2D_MultiRes_UNet.py ├── 2D_attentionunet.py ├── 2D_r2unet.py ├── 2D_swinunet.py ├── 2D_swinunet3+.py ├── 2D_transunet.py ├── 2D_u2net.py ├── 2D_unet++.py ├── 2D_unet.py ├── 2D_unet3+.py ├── 2D_vnet.py ├── 3D_Dense_UNet.py ├── 3D_Residual_UNet.py ├── 3D_UNet.py ├── LICENSE ├── Process_Data_for_2D_NoisyLabel_Spine.py ├── Process_Data_for_2D_SparseLabel_numpy.py ├── README.md ├── attention_module_2D.py ├── attention_module_3D.py ├── evaluation.py ├── imgs ├── noisylabel.png └── sparselabel.png ├── keras_unet_collection ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── _backbone_zoo.cpython-38.pyc │ ├── _model_att_unet_2d.cpython-38.pyc │ ├── _model_r2_unet_2d.cpython-38.pyc │ ├── _model_resunet_a_2d.cpython-38.pyc │ ├── _model_swin_unet_2d.cpython-38.pyc │ ├── _model_transunet_2d.cpython-38.pyc │ ├── _model_u2net_2d.cpython-38.pyc │ ├── _model_unet_2d.cpython-38.pyc │ ├── _model_unet_3plus_2d.cpython-38.pyc │ ├── _model_unet_plus_2d.cpython-38.pyc │ ├── _model_vnet_2d.cpython-38.pyc │ ├── activations.cpython-38.pyc │ ├── base.cpython-38.pyc │ ├── layer_utils.cpython-38.pyc │ ├── losses.cpython-38.pyc │ ├── models.cpython-38.pyc │ ├── transformer_layers.cpython-38.pyc │ └── utils.cpython-38.pyc ├── _backbone_zoo.py ├── _model_att_unet_2d.py ├── _model_r2_unet_2d.py ├── _model_resunet_a_2d.py ├── _model_swin_unet_2d.py ├── _model_transunet_2d.py ├── _model_u2net_2d.py ├── _model_unet_2d.py ├── _model_unet_3plus_2d.py ├── _model_unet_plus_2d.py ├── _model_vnet_2d.py ├── activations.py ├── backbones.py ├── base.py ├── layer_utils.py ├── losses.py ├── models.py ├── transformer_layers.py └── utils.py └── segmentation_models ├── __init__.py ├── __version__.py ├── backbones ├── __init__.py ├── backbones_factory.py ├── inception_resnet_v2.py └── inception_v3.py ├── base ├── __init__.py ├── functional.py └── objects.py ├── losses.py ├── metrics.py ├── models ├── __init__.py ├── _common_blocks.py ├── _utils.py ├── fpn.py ├── linknet.py ├── pspnet.py └── unet.py └── utils.py /2D_Dense_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_Dense_UNet.py -------------------------------------------------------------------------------- /2D_FPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_FPN.py -------------------------------------------------------------------------------- /2D_Linknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_Linknet.py -------------------------------------------------------------------------------- /2D_MultiRes_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_MultiRes_UNet.py -------------------------------------------------------------------------------- /2D_attentionunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_attentionunet.py -------------------------------------------------------------------------------- /2D_r2unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_r2unet.py -------------------------------------------------------------------------------- /2D_swinunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_swinunet.py -------------------------------------------------------------------------------- /2D_swinunet3+.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_swinunet3+.py -------------------------------------------------------------------------------- /2D_transunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_transunet.py -------------------------------------------------------------------------------- /2D_u2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_u2net.py -------------------------------------------------------------------------------- /2D_unet++.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_unet++.py -------------------------------------------------------------------------------- /2D_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_unet.py -------------------------------------------------------------------------------- /2D_unet3+.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_unet3+.py -------------------------------------------------------------------------------- /2D_vnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/2D_vnet.py -------------------------------------------------------------------------------- /3D_Dense_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/3D_Dense_UNet.py -------------------------------------------------------------------------------- /3D_Residual_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/3D_Residual_UNet.py -------------------------------------------------------------------------------- /3D_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/3D_UNet.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/LICENSE -------------------------------------------------------------------------------- /Process_Data_for_2D_NoisyLabel_Spine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/Process_Data_for_2D_NoisyLabel_Spine.py -------------------------------------------------------------------------------- /Process_Data_for_2D_SparseLabel_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/Process_Data_for_2D_SparseLabel_numpy.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/README.md -------------------------------------------------------------------------------- /attention_module_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/attention_module_2D.py -------------------------------------------------------------------------------- /attention_module_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/attention_module_3D.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/evaluation.py -------------------------------------------------------------------------------- /imgs/noisylabel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/imgs/noisylabel.png -------------------------------------------------------------------------------- /imgs/sparselabel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/imgs/sparselabel.png -------------------------------------------------------------------------------- /keras_unet_collection/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_backbone_zoo.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_backbone_zoo.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_att_unet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_att_unet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_r2_unet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_r2_unet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_resunet_a_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_resunet_a_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_swin_unet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_swin_unet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_transunet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_transunet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_u2net_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_u2net_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_unet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_unet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_unet_3plus_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_unet_3plus_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_unet_plus_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_unet_plus_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/_model_vnet_2d.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/_model_vnet_2d.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/activations.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/activations.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/base.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/base.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/layer_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/layer_utils.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/losses.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/losses.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/models.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/models.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/transformer_layers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/transformer_layers.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /keras_unet_collection/_backbone_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_backbone_zoo.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_att_unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_att_unet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_r2_unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_r2_unet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_resunet_a_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_resunet_a_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_swin_unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_swin_unet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_transunet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_transunet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_u2net_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_u2net_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_unet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_unet_3plus_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_unet_3plus_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_unet_plus_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_unet_plus_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/_model_vnet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/_model_vnet_2d.py -------------------------------------------------------------------------------- /keras_unet_collection/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/activations.py -------------------------------------------------------------------------------- /keras_unet_collection/backbones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/backbones.py -------------------------------------------------------------------------------- /keras_unet_collection/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/base.py -------------------------------------------------------------------------------- /keras_unet_collection/layer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/layer_utils.py -------------------------------------------------------------------------------- /keras_unet_collection/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/losses.py -------------------------------------------------------------------------------- /keras_unet_collection/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/models.py -------------------------------------------------------------------------------- /keras_unet_collection/transformer_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/transformer_layers.py -------------------------------------------------------------------------------- /keras_unet_collection/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/keras_unet_collection/utils.py -------------------------------------------------------------------------------- /segmentation_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/__init__.py -------------------------------------------------------------------------------- /segmentation_models/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/__version__.py -------------------------------------------------------------------------------- /segmentation_models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation_models/backbones/backbones_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/backbones/backbones_factory.py -------------------------------------------------------------------------------- /segmentation_models/backbones/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/backbones/inception_resnet_v2.py -------------------------------------------------------------------------------- /segmentation_models/backbones/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/backbones/inception_v3.py -------------------------------------------------------------------------------- /segmentation_models/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/base/__init__.py -------------------------------------------------------------------------------- /segmentation_models/base/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/base/functional.py -------------------------------------------------------------------------------- /segmentation_models/base/objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/base/objects.py -------------------------------------------------------------------------------- /segmentation_models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/losses.py -------------------------------------------------------------------------------- /segmentation_models/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/metrics.py -------------------------------------------------------------------------------- /segmentation_models/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation_models/models/_common_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/_common_blocks.py -------------------------------------------------------------------------------- /segmentation_models/models/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/_utils.py -------------------------------------------------------------------------------- /segmentation_models/models/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/fpn.py -------------------------------------------------------------------------------- /segmentation_models/models/linknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/linknet.py -------------------------------------------------------------------------------- /segmentation_models/models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/pspnet.py -------------------------------------------------------------------------------- /segmentation_models/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/models/unet.py -------------------------------------------------------------------------------- /segmentation_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziyangwang007/VIT4UNet/HEAD/segmentation_models/utils.py --------------------------------------------------------------------------------