Home

Awesome

XBound-Former: Toward Cross-scale Boundary Modeling in Transformers

Introduction

This is an official release of the paper XBound-Former: Toward Cross-scale Boundary Modeling in Transformers, including the network implementation and the training scripts.

[XBound-Former: Toward Cross-scale Boundary Modeling in Transformers]<br/> Jiacheng Wang, Fei Chen, Yuxi Ma, Liansheng Wang, Zhaodong Fei, Jianwei Shuai, Xiangdong Tang, Qichao Zhou, Jing Qin <br/> In: Transactions on Medical Imaging (TMI), 2023 <br/> [arXiv][Bibetex]

<div align="center" border=> <img src=frame.jpg width="700" > </div>

News

Code List

For more details or any questions, please feel easy to contact us by email (jiachengw@stu.xmu.edu.cn).

Usage

Dataset

Please download the dataset from ISIC challenge and PH2 website.

Pre-processing

Please run:

$ python utils/resize.py

You need to change the File Path to your own and select the correct function.

Training

Please run:

$ python src/train.py

You need to change the File Path to your own and select the correct function.

Testing

Download the pretrained weight for ISCI-2016&$ph^2$ dataset from Google Drive and move to the logger dir.

Then, please run:

$ python src/test.py

Result

The ISIC-2016&$ph^2$ dataset:

<div align="center" border=> <img src=isic2016.png width="700" > </div>

Citation

If you find XBound-Former useful in your research, please consider citing:

@article{wang2023xbound,
  title={XBound-Former: Toward Cross-scale Boundary Modeling in Transformers},
  author={Wang, Jiacheng and Chen, Fei and Ma, Yuxi and Wang, Liansheng and Fei, Zhaodong and Shuai, Jianwei and Tang, Xiangdong and Zhou, Qichao and Qin, Jing},
  journal={IEEE Transactions on Medical Imaging},
  year={2023},
  publisher={IEEE}
}

and the prior work, BAT, as:

@inproceedings{wang2021boundary,
  title={Boundary-Aware Transformers for Skin Lesion Segmentation},
  author={Wang, Jiacheng and Wei, Lan and Wang, Liansheng and Zhou, Qichao and Zhu, Lei and Qin, Jing},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={206--216},
  year={2021},
  organization={Springer}
}