Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 1.15 KB

README.md

File metadata and controls

32 lines (23 loc) · 1.15 KB

TFormer

MIT Licence

Introduction

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

Our Network Structure

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: