├── .gitignore ├── LICENSE ├── README.md ├── architectures ├── __init__.py ├── build_architecture.py └── segformer3d.py ├── augmentations ├── __init__.py └── augmentations.py ├── cvprw_poster.pdf ├── data ├── brats2017_seg │ ├── brats2017_raw_data │ │ ├── brats2017_seg_preprocess.py │ │ └── datameta_generator │ │ │ ├── create_train_val_kfold_csv.py │ │ │ ├── create_train_val_test_csv.py │ │ │ ├── nnformer_train_test_split.py │ │ │ ├── train.csv │ │ │ ├── train_fold_1.csv │ │ │ ├── train_fold_2.csv │ │ │ ├── train_fold_3.csv │ │ │ ├── train_fold_4.csv │ │ │ ├── train_fold_5.csv │ │ │ ├── validation.csv │ │ │ ├── validation_fold_1.csv │ │ │ ├── validation_fold_2.csv │ │ │ ├── validation_fold_3.csv │ │ │ ├── validation_fold_4.csv │ │ │ └── validation_fold_5.csv │ ├── train.csv │ └── validation.csv └── brats2021_seg │ ├── brats2021_raw_data │ ├── brats2021_seg_preprocess.py │ └── datameta_generator │ │ ├── create_train_val_kfold_csv.py │ │ ├── create_train_val_test_csv.py │ │ ├── create_train_val_test_csv_v2.py │ │ ├── train_v2.csv │ │ └── validation_v2.csv │ ├── train.csv │ └── validation.csv ├── dataloaders ├── __init__.py ├── brats2017_seg.py ├── brats2021_seg.py └── build_dataset.py ├── experiments └── brats_2017 │ ├── best_brats_2017_exp_dice_82.07 │ ├── config.yaml │ ├── run_experiment.py │ └── single_gpu_accelerate.yaml │ └── template_experiment │ ├── config.yaml │ ├── gpu_accelerate.yaml │ └── run_experiment.py ├── losses ├── __init__.py └── losses.py ├── metrics └── segmentation_metrics.py ├── notebooks └── model_profiler.ipynb ├── optimizers ├── __init__.py ├── optimizers.py └── schedulers.py ├── requirements.txt ├── resources ├── acdc_quant.png ├── acdc_segformer_3D.png ├── brats_plot.png ├── brats_quant.png ├── brats_segformer_3D.png ├── param_count_table.PNG ├── segformer_3D.png ├── synapse_quant.png └── synapse_segformer_3D.png └── train_scripts ├── __init__.py ├── trainer_ddp.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/README.md -------------------------------------------------------------------------------- /architectures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /architectures/build_architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/architectures/build_architecture.py -------------------------------------------------------------------------------- /architectures/segformer3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/architectures/segformer3d.py -------------------------------------------------------------------------------- /augmentations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /augmentations/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/augmentations/augmentations.py -------------------------------------------------------------------------------- /cvprw_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/cvprw_poster.pdf -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/brats2017_seg_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/brats2017_seg_preprocess.py -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/create_train_val_kfold_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/create_train_val_kfold_csv.py -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/create_train_val_test_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/create_train_val_test_csv.py -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/nnformer_train_test_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/nnformer_train_test_split.py -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_1.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_2.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_3.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_4.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_4.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_5.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/train_fold_5.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_1.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_2.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_3.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_4.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_4.csv -------------------------------------------------------------------------------- /data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_5.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/brats2017_raw_data/datameta_generator/validation_fold_5.csv -------------------------------------------------------------------------------- /data/brats2017_seg/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/train.csv -------------------------------------------------------------------------------- /data/brats2017_seg/validation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2017_seg/validation.csv -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/brats2021_seg_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/brats2021_seg_preprocess.py -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_kfold_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_kfold_csv.py -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_test_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_test_csv.py -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_test_csv_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/datameta_generator/create_train_val_test_csv_v2.py -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/datameta_generator/train_v2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/datameta_generator/train_v2.csv -------------------------------------------------------------------------------- /data/brats2021_seg/brats2021_raw_data/datameta_generator/validation_v2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/brats2021_raw_data/datameta_generator/validation_v2.csv -------------------------------------------------------------------------------- /data/brats2021_seg/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/train.csv -------------------------------------------------------------------------------- /data/brats2021_seg/validation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/data/brats2021_seg/validation.csv -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataloaders/brats2017_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/dataloaders/brats2017_seg.py -------------------------------------------------------------------------------- /dataloaders/brats2021_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/dataloaders/brats2021_seg.py -------------------------------------------------------------------------------- /dataloaders/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/dataloaders/build_dataset.py -------------------------------------------------------------------------------- /experiments/brats_2017/best_brats_2017_exp_dice_82.07/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/best_brats_2017_exp_dice_82.07/config.yaml -------------------------------------------------------------------------------- /experiments/brats_2017/best_brats_2017_exp_dice_82.07/run_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/best_brats_2017_exp_dice_82.07/run_experiment.py -------------------------------------------------------------------------------- /experiments/brats_2017/best_brats_2017_exp_dice_82.07/single_gpu_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/best_brats_2017_exp_dice_82.07/single_gpu_accelerate.yaml -------------------------------------------------------------------------------- /experiments/brats_2017/template_experiment/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/template_experiment/config.yaml -------------------------------------------------------------------------------- /experiments/brats_2017/template_experiment/gpu_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/template_experiment/gpu_accelerate.yaml -------------------------------------------------------------------------------- /experiments/brats_2017/template_experiment/run_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/experiments/brats_2017/template_experiment/run_experiment.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/losses/losses.py -------------------------------------------------------------------------------- /metrics/segmentation_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/metrics/segmentation_metrics.py -------------------------------------------------------------------------------- /notebooks/model_profiler.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/notebooks/model_profiler.ipynb -------------------------------------------------------------------------------- /optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /optimizers/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/optimizers/optimizers.py -------------------------------------------------------------------------------- /optimizers/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/optimizers/schedulers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/acdc_quant.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/acdc_quant.png -------------------------------------------------------------------------------- /resources/acdc_segformer_3D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/acdc_segformer_3D.png -------------------------------------------------------------------------------- /resources/brats_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/brats_plot.png -------------------------------------------------------------------------------- /resources/brats_quant.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/brats_quant.png -------------------------------------------------------------------------------- /resources/brats_segformer_3D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/brats_segformer_3D.png -------------------------------------------------------------------------------- /resources/param_count_table.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/param_count_table.PNG -------------------------------------------------------------------------------- /resources/segformer_3D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/segformer_3D.png -------------------------------------------------------------------------------- /resources/synapse_quant.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/synapse_quant.png -------------------------------------------------------------------------------- /resources/synapse_segformer_3D.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/resources/synapse_segformer_3D.png -------------------------------------------------------------------------------- /train_scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_scripts/trainer_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/train_scripts/trainer_ddp.py -------------------------------------------------------------------------------- /train_scripts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OSUPCVLab/SegFormer3D/HEAD/train_scripts/utils.py --------------------------------------------------------------------------------