NITR098/AttSwinUNet

Files not found

Eugenieroo opened this issue · 1 comments

Hi, I'm trying to run your model.

At "Prepare data and pretrained model" part, in the 3rd point I need the "Train_Skin_Lesion_Segmentation.py" file. I don't find it in the repository.
Also, I'm having issues with this instruction: from networks.attention_swin_unet import SwinAttentionUnet as ViT_seg
I have downloaded all the requirements in the "requirements.txt", as you indicate, and I don't know how to fix this.

Thanks for your help : )

you do not need this file I remove from read me please just run train.py after preparing dataset.