/ConTrans

This repository contains the official code of ConTrans: Improving Transformer with Convolutional Attention for Medical Image Segmentation

Primary LanguagePython

ConTrans

This repository contains the official code of ConTrans: Improving Transformer with Convolutional Attention for Medical Image Segmentation

Requirements

  • mmcv==1.7.1
  • mmdet==2.28.2

Pretrained models

Download Swin-B from Swin-Transformer or Baidu Pan (ji2g) to ./checkpoints

Citation

Please consider citing us if you find this work helpful:

@inproceedings{lin2022contrans,
  title={ConTrans: ConTrans: Improving Transformer with Convolutional Attention for Medical Image Segmentation},
  author={Lin, Ailiang and Xu, Jiayu and Li, Jinxing and Lu, Guangming},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={297--307},
  year={2022},
  organization={Springer}
}

Questions

Please drop an email to tianbaoge24@gmail.com