Home

Awesome

Boundary-aware Transformers for Skin Lesion Segmentation

Introduction

This is an official release of the paper Boundary-aware Transformers for Skin Lesion Segmentation.

Boundary-aware Transformers for Skin Lesion Segmentation, <br/> Jiacheng Wang, Lan Wei, Liansheng Wang, Qichao Zhou, Lei Zhu, Jing Qin <br/> In: Medical Image Computing and Computer Assisted Intervention (MICCAI), 2021 <br/> [arXiv][Bibetex]

<div align="center" border=> <img src=framework.jpg width="400" > </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 src/process_resize.py
$ python src/process_point.py

You need to change the File Path to your own.

Point Maps

For your convenience, we release the processed maps and the dataset division.

Please download them from Baidu Disk (code:kmqr) or Google Drive

The file names are equal to the original image names.

Training

Testing

Download the pretrained weight for PH2 dataset from Google Drive.

$ python test.py --dataset isic2016

Result

MethodDiceIoUHD95ASSD
Lee et al.0.9180.843--
BAT (paper)0.9210.858--
<!-- BAT (val.)| 0.918 | 0.854 | **6.664** | **0.923** | --> <!-- BAT (w/o val.)| **0.923** | **0.864** | 7.515 | 1.043 | -->

Citation

If you find BAT useful in your research, please consider citing:

@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}
}