BATFormer

This repo is the official implementation for:
BATFormer: Towards Boundary-Aware Lightweight Transformer for Efficient Medical Image Segmentation.
(The details of our BATFormer can be found at the models directory in this repo or in the paper.)

Requirements

• python 3.6
• pytorch 1.8.0
• torchvision 0.9.0
• more details please see the requirements.txt

Datasets

• The ACDC dataset could be acquired from here. The slice-level ACDC dataset preprocessed by us can be downloaded from here.
• The ISIC 2018 dataset could be acquired from here.

Training

Commands for training

python train.py

Testing

Commands for testing

python test.py

References

1. vit-pytorch
2. MedT