/ConvFormer

Primary LanguagePythonMIT LicenseMIT

ConvFormer

This repo is the official implementation for:
[MICCAI2023] ConvFormer: Plug-and-Play CNN-Style Transformers for Improving Medical Image Segmentation.
(The details of our ConvFormer can be found at the models directory in this repo or in the paper. We take SETR for example.)

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