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.
<div align="center" border=> <img src=frame.jpg width="700" > </div>[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]
News
- [1/11 2022] This paper has been accepted to TMI.
- [5/27 2022] We have released the training scripts.
- [5/19 2022] We have created this repo.
Code List
- Network
- Pre-processing
- Training Codes
- Pretrained Weights
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}
}