/TFCNs

This repository includes the official project of TFCNs, presented in our paper: TFCNs: A CNN-Transformer Hybrid Network for Medical Image Segmentation

Primary LanguagePythonApache License 2.0Apache-2.0

TFCNs

This repository includes the official project of TFCNs, presented in our paper: TFCNs: A CNN-Transformer Hybrid Network for Medical Image Segmentation

image

paper link: https://arxiv.org/abs/2207.03450

Gmail: dihanli@stu.xmu.edu.cn

Please contact me if you need the dataset.

More details of code & readme will come soon!

Usage

model/ : save for the model you have train

networks/ : all the component that construct our TFCNs

preprocess.py : simple data augumentation

train_utils.py : some tools used for training

utils.py : some tools used for testing

you can run the train.py and test.py for training and testing.

Environment

Please prepare an environment with python=3.7, and then use the command "pip install -r requirements.txt" for the dependencies.

Citation

@article{li2022TFCNs,
  title={TFCNs: A CNN-Transformer Hybrid Network for Medical Image Segmentation},
  author={Li, Zihan and Li, Dihan and Xu, Cangbai and Wang, Weice and Hong, Qingqi and Li, Qingde and Tian, Jie},
  journal={arXiv preprint arXiv:2207.03450},
  year={2022}
}