TFormer

Introduction

The official implementation of "TFormer: A throughout fusion transformer for multi-modal skin lesion diagnosis"

Enviroments

Windows/Linux both support
python 3.9
PyTorch 1.12.1
torchvision

Prepare dataset

Please at first download datasets Derm7pt and then download the pretrained model of swin-tiny on ImageNet-1k from github. Save the model into the folder "./models/swin_transformer".

Run details

To train our TFormer, run:

python train.py --dir_release "your dataset path" --epochs 100 --batch_size 32 --learning_rate 1e-4 --cuda True

License

This project is licensed under the MIT License. See LICENSE for details

Acknowledgement

Our code borrows a lot from:

Swin-Transformer
Derm7pt