Official PyTorch Implementation of EoFormer
EoFormer: Edge-oriented Transformer for Brain Tumor Segmentation paper
- Clone the repository
git clone https://github.com/sd0809/EoFormer.git
- Create a conda environment and install the required packages
conda create -n EoFormer python==3.8
conda activate EoFormer
pip install -r requirements.txt
Put the BraTS 2020 dataset in data_path
You can download BraTS 2020 dataset in here
BraTS2020
| -- BraTS2020_TrainingData
| -- BraTS2020_ValidationData
python main.py --data_path <data path> --args ……
or sh run.sh
@inproceedings{she2023eoformer,
title={EoFormer: Edge-Oriented Transformer for Brain Tumor Segmentation},
author={She, Dong and Zhang, Yueyi and Zhang, Zheyu and Li, Hebei and Yan, Zihan and Sun, Xiaoyan},
booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
pages={333--343},
year={2023},
organization={Springer}
}
If you find our project useful, please consider starring it : )