Skip to content

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

License

Notifications You must be signed in to change notification settings

HUANGLIZI/TFCNs

Repository files navigation

TFCNs (ICANN 2022 Oral)

This repository includes the official project of TFCNs, presented in our paper: TFCNs: A CNN-Transformer Hybrid Network for Medical Image Segmentation , which is accepted by ICANN 2022 (International Conference on Artificial Neural Networks). image

paper link: https://arxiv.org/abs/2207.03450 or https://doi.org/10.1007/978-3-031-15937-4_65

Email: [email protected]

Please contact dihan or me if you need the further help.

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

@inproceedings{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},
  booktitle={International Conference on Artificial Neural Networks},
  pages={781--792},
  year={2022},
  organization={Springer}
}

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages