Skip to content

Latest commit

 

History

History
48 lines (37 loc) · 1.27 KB

README.md

File metadata and controls

48 lines (37 loc) · 1.27 KB

EoFormer

Official PyTorch Implementation of EoFormer

EoFormer: Edge-oriented Transformer for Brain Tumor Segmentation paper

pipeline pipeline

Enviroments

  1. Clone the repository

git clone https://github.com/sd0809/EoFormer.git

  1. Create a conda environment and install the required packages
conda create -n EoFormer python==3.8
conda activate EoFormer
pip install -r requirements.txt

Data

Put the BraTS 2020 dataset in data_path

You can download BraTS 2020 dataset in here

BraTS2020
| -- BraTS2020_TrainingData
| -- BraTS2020_ValidationData

Train & Test

python main.py --data_path <data path> --args …… or sh run.sh

Citation

@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 : )