nitr098 / attswinunet Goto Github PK

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 : )

#4
It is a great job.
But your code is really chaos, i wonder if you upload the wrong vision?
Or just ''Pretend to be Open Source''
Best wish~

I had this error : /usr/bin/python3: can't find 'main' module in '/content/drive/MyDrive/AttSwinUNet-main/AttSwinUNet-main/model' did I miss something ?

I can‘t find the utils files，when I run the rode，it tell me it can't find the load_pretrain,can you help me?

There is no py file in your commit

Hi, I run train.py and got an error from the code 'from utils import load_pretrain'. I tried the pip install utils module from python but still got the same error. Is file 'utils.py' exist?