The codes for the work "HTC-Net:
- [Get pre-trained model in this link] (https://drive.google.com/drive/folders/1UC3XOoezeum0uck4KBVGa8osahs6rKUY?usp=sharing): Put pretrained Swin-T into folder "pretrained_ckpt/" and create dir 'chechpoint','test_log' in the root path.
- You can go to https://challenge.isic-archive.com/data/#2017 to acquire the ISIC-2017 dataset.
- You can go to https://paperswithcode.com/dataset/kvasir to acquire the Kvasir-SEG dataset.
- Please prepare an environment with python=3.8, and then use the command "pip install -r requirements.txt" for the dependencies.
-
Run the train script on the ISIC-2017 and the COVID-QU-Ex dataset. The batch size we used is 8. If you do not have enough GPU memory, the bacth size can be reduced to 4 or 6 to save memory. For more information, contact [email protected].
-
Train
python train.py
- Test
python test.py