├── .gitignore ├── LICENSE ├── README.md ├── config ├── acdc │ ├── attention_unet_2d.yaml │ ├── attention_unet_3d.yaml │ ├── medformer_2d.yaml │ ├── medformer_3d.yaml │ ├── nnformer_3d.yaml │ ├── resunet_2d.yaml │ ├── resunet_3d.yaml │ ├── swinunet_2d.yaml │ ├── transunet_2d.yaml │ ├── unet++_2d.yaml │ ├── unet++_3d.yaml │ ├── unet_2d.yaml │ ├── unet_3d.yaml │ ├── unetr_3d.yaml │ ├── vnet_3d.yaml │ └── vtunet_3d.yaml ├── amos_ct │ ├── attention_unet_3d.yaml │ ├── medformer_3d.yaml │ ├── resunet_3d.yaml │ └── vtunet_3d.yaml ├── amos_mr │ ├── attention_unet_3d.yaml │ ├── medformer_3d.yaml │ ├── resunet_3d.yaml │ └── vtunet_3d.yaml ├── bcv │ ├── attention_unet_3d.yaml │ ├── medformer_3d.yaml │ ├── nnformer_3d.yaml │ ├── resunet_3d.yaml │ ├── swin_unetr_3d.yaml │ ├── unetr_3d.yaml │ └── vtunet_3d.yaml ├── kits │ ├── attention_unet_3d.yaml │ ├── medformer_3d.yaml │ ├── nnformer_3d.yaml │ ├── resunet_3d.yaml │ ├── swin_unetr_3d.yaml │ ├── unetr_3d.yaml │ └── vtunet_3d.yaml └── lits │ ├── attention_unet_3d.yaml │ ├── medformer_3d.yaml │ ├── nnformer_3d.yaml │ ├── resunet_3d.yaml │ ├── swin_unetr_3d.yaml │ ├── unetr_3d.yaml │ └── vtunet_3d.yaml ├── dataset_conversion ├── __init__.py ├── acdc_2d.py ├── acdc_3d.py ├── amos_3d.py ├── bcv_3d.py ├── kits_3d.py ├── lits_3d.py └── utils.py ├── docs ├── change.md └── tutorial.md ├── inference ├── inference2d.py ├── inference3d.py └── utils.py ├── metric ├── lookup_tables.py ├── metrics.py └── utils.py ├── model ├── __init__.py ├── dim2 │ ├── __init__.py │ ├── attention_unet.py │ ├── attention_unet_utils.py │ ├── conv_layers.py │ ├── dual_attention_unet.py │ ├── dual_attention_utils.py │ ├── medformer.py │ ├── medformer_utils.py │ ├── swin_unet.py │ ├── trans_layers.py │ ├── transunet.py │ ├── unet.py │ ├── unet_utils.py │ ├── unetpp.py │ └── utils.py ├── dim3 │ ├── __init__.py │ ├── attention_unet.py │ ├── attention_unet_utils.py │ ├── conv_layers.py │ ├── medformer.py │ ├── medformer_utils.py │ ├── medformer_utils_v2.py │ ├── nnformer.py │ ├── nnformer_utils.py │ ├── swin_unetr.py │ ├── trans_layers.py │ ├── unet.py │ ├── unet_utils.py │ ├── unetpp.py │ ├── unetr.py │ ├── utils.py │ ├── vnet.py │ ├── vtunet.py │ └── vtunet_utils.py └── utils.py ├── prediction.py ├── requirements.txt ├── train.py ├── train_ddp.py ├── training ├── __init__.py ├── augmentation.py ├── dataset │ ├── __init__.py │ ├── dim2 │ │ ├── __init__.py │ │ └── dataset_acdc.py │ ├── dim3 │ │ ├── dataset_acdc.py │ │ ├── dataset_amos_ct.py │ │ ├── dataset_amos_mr.py │ │ ├── dataset_bcv.py │ │ ├── dataset_kits.py │ │ └── dataset_lits.py │ └── utils.py ├── losses.py ├── utils.py └── validation.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/README.md -------------------------------------------------------------------------------- /config/acdc/attention_unet_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/attention_unet_2d.yaml -------------------------------------------------------------------------------- /config/acdc/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/acdc/medformer_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/medformer_2d.yaml -------------------------------------------------------------------------------- /config/acdc/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/medformer_3d.yaml -------------------------------------------------------------------------------- /config/acdc/nnformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/nnformer_3d.yaml -------------------------------------------------------------------------------- /config/acdc/resunet_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/resunet_2d.yaml -------------------------------------------------------------------------------- /config/acdc/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/resunet_3d.yaml -------------------------------------------------------------------------------- /config/acdc/swinunet_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/swinunet_2d.yaml -------------------------------------------------------------------------------- /config/acdc/transunet_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/transunet_2d.yaml -------------------------------------------------------------------------------- /config/acdc/unet++_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/unet++_2d.yaml -------------------------------------------------------------------------------- /config/acdc/unet++_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/unet++_3d.yaml -------------------------------------------------------------------------------- /config/acdc/unet_2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/unet_2d.yaml -------------------------------------------------------------------------------- /config/acdc/unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/unet_3d.yaml -------------------------------------------------------------------------------- /config/acdc/unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/unetr_3d.yaml -------------------------------------------------------------------------------- /config/acdc/vnet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/vnet_3d.yaml -------------------------------------------------------------------------------- /config/acdc/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/acdc/vtunet_3d.yaml -------------------------------------------------------------------------------- /config/amos_ct/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_ct/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/amos_ct/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_ct/medformer_3d.yaml -------------------------------------------------------------------------------- /config/amos_ct/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_ct/resunet_3d.yaml -------------------------------------------------------------------------------- /config/amos_ct/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_ct/vtunet_3d.yaml -------------------------------------------------------------------------------- /config/amos_mr/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_mr/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/amos_mr/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_mr/medformer_3d.yaml -------------------------------------------------------------------------------- /config/amos_mr/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_mr/resunet_3d.yaml -------------------------------------------------------------------------------- /config/amos_mr/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/amos_mr/vtunet_3d.yaml -------------------------------------------------------------------------------- /config/bcv/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/bcv/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/medformer_3d.yaml -------------------------------------------------------------------------------- /config/bcv/nnformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/nnformer_3d.yaml -------------------------------------------------------------------------------- /config/bcv/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/resunet_3d.yaml -------------------------------------------------------------------------------- /config/bcv/swin_unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/swin_unetr_3d.yaml -------------------------------------------------------------------------------- /config/bcv/unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/unetr_3d.yaml -------------------------------------------------------------------------------- /config/bcv/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/bcv/vtunet_3d.yaml -------------------------------------------------------------------------------- /config/kits/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/kits/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/medformer_3d.yaml -------------------------------------------------------------------------------- /config/kits/nnformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/nnformer_3d.yaml -------------------------------------------------------------------------------- /config/kits/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/resunet_3d.yaml -------------------------------------------------------------------------------- /config/kits/swin_unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/swin_unetr_3d.yaml -------------------------------------------------------------------------------- /config/kits/unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/unetr_3d.yaml -------------------------------------------------------------------------------- /config/kits/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/kits/vtunet_3d.yaml -------------------------------------------------------------------------------- /config/lits/attention_unet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/attention_unet_3d.yaml -------------------------------------------------------------------------------- /config/lits/medformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/medformer_3d.yaml -------------------------------------------------------------------------------- /config/lits/nnformer_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/nnformer_3d.yaml -------------------------------------------------------------------------------- /config/lits/resunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/resunet_3d.yaml -------------------------------------------------------------------------------- /config/lits/swin_unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/swin_unetr_3d.yaml -------------------------------------------------------------------------------- /config/lits/unetr_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/unetr_3d.yaml -------------------------------------------------------------------------------- /config/lits/vtunet_3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/config/lits/vtunet_3d.yaml -------------------------------------------------------------------------------- /dataset_conversion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset_conversion/acdc_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/acdc_2d.py -------------------------------------------------------------------------------- /dataset_conversion/acdc_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/acdc_3d.py -------------------------------------------------------------------------------- /dataset_conversion/amos_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/amos_3d.py -------------------------------------------------------------------------------- /dataset_conversion/bcv_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/bcv_3d.py -------------------------------------------------------------------------------- /dataset_conversion/kits_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/kits_3d.py -------------------------------------------------------------------------------- /dataset_conversion/lits_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/lits_3d.py -------------------------------------------------------------------------------- /dataset_conversion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/dataset_conversion/utils.py -------------------------------------------------------------------------------- /docs/change.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/docs/change.md -------------------------------------------------------------------------------- /docs/tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/docs/tutorial.md -------------------------------------------------------------------------------- /inference/inference2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/inference/inference2d.py -------------------------------------------------------------------------------- /inference/inference3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/inference/inference3d.py -------------------------------------------------------------------------------- /inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/inference/utils.py -------------------------------------------------------------------------------- /metric/lookup_tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/metric/lookup_tables.py -------------------------------------------------------------------------------- /metric/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/metric/metrics.py -------------------------------------------------------------------------------- /metric/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/metric/utils.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/dim2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/__init__.py -------------------------------------------------------------------------------- /model/dim2/attention_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/attention_unet.py -------------------------------------------------------------------------------- /model/dim2/attention_unet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/attention_unet_utils.py -------------------------------------------------------------------------------- /model/dim2/conv_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/conv_layers.py -------------------------------------------------------------------------------- /model/dim2/dual_attention_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/dual_attention_unet.py -------------------------------------------------------------------------------- /model/dim2/dual_attention_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/dual_attention_utils.py -------------------------------------------------------------------------------- /model/dim2/medformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/medformer.py -------------------------------------------------------------------------------- /model/dim2/medformer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/medformer_utils.py -------------------------------------------------------------------------------- /model/dim2/swin_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/swin_unet.py -------------------------------------------------------------------------------- /model/dim2/trans_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/trans_layers.py -------------------------------------------------------------------------------- /model/dim2/transunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/transunet.py -------------------------------------------------------------------------------- /model/dim2/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/unet.py -------------------------------------------------------------------------------- /model/dim2/unet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/unet_utils.py -------------------------------------------------------------------------------- /model/dim2/unetpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/unetpp.py -------------------------------------------------------------------------------- /model/dim2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim2/utils.py -------------------------------------------------------------------------------- /model/dim3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/__init__.py -------------------------------------------------------------------------------- /model/dim3/attention_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/attention_unet.py -------------------------------------------------------------------------------- /model/dim3/attention_unet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/attention_unet_utils.py -------------------------------------------------------------------------------- /model/dim3/conv_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/conv_layers.py -------------------------------------------------------------------------------- /model/dim3/medformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/medformer.py -------------------------------------------------------------------------------- /model/dim3/medformer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/medformer_utils.py -------------------------------------------------------------------------------- /model/dim3/medformer_utils_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/medformer_utils_v2.py -------------------------------------------------------------------------------- /model/dim3/nnformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/nnformer.py -------------------------------------------------------------------------------- /model/dim3/nnformer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/nnformer_utils.py -------------------------------------------------------------------------------- /model/dim3/swin_unetr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/swin_unetr.py -------------------------------------------------------------------------------- /model/dim3/trans_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/trans_layers.py -------------------------------------------------------------------------------- /model/dim3/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/unet.py -------------------------------------------------------------------------------- /model/dim3/unet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/unet_utils.py -------------------------------------------------------------------------------- /model/dim3/unetpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/unetpp.py -------------------------------------------------------------------------------- /model/dim3/unetr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/unetr.py -------------------------------------------------------------------------------- /model/dim3/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/utils.py -------------------------------------------------------------------------------- /model/dim3/vnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/vnet.py -------------------------------------------------------------------------------- /model/dim3/vtunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/vtunet.py -------------------------------------------------------------------------------- /model/dim3/vtunet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/dim3/vtunet_utils.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/model/utils.py -------------------------------------------------------------------------------- /prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/prediction.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/train.py -------------------------------------------------------------------------------- /train_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/train_ddp.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/augmentation.py -------------------------------------------------------------------------------- /training/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/dataset/dim2/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /training/dataset/dim2/dataset_acdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim2/dataset_acdc.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_acdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_acdc.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_amos_ct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_amos_ct.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_amos_mr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_amos_mr.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_bcv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_bcv.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_kits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_kits.py -------------------------------------------------------------------------------- /training/dataset/dim3/dataset_lits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/dim3/dataset_lits.py -------------------------------------------------------------------------------- /training/dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/dataset/utils.py -------------------------------------------------------------------------------- /training/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/losses.py -------------------------------------------------------------------------------- /training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/utils.py -------------------------------------------------------------------------------- /training/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/training/validation.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhygao/CBIM-Medical-Image-Segmentation/HEAD/utils.py --------------------------------------------------------------------------------