Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix ISIC dataloader, shape error in UNet & questions about paper #1

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

halleewong
Copy link

Hi,

Thanks for sharing this code! Posting here because it seems like there's no "issues" tab in your repo.

I am very interested in comparing to your model. Would you be able to release the pre-trained weights? It would make it much easier to compare and the pre-trained weights are necessary to run the val.py examples in your readme.

I tried running the train.py example using the provided ISIC dataloader (fixed some bugs in 1cb9712 to get it working) but I encountered a shape error with the recommended unet backbone. The image embeddings output by the unet encoder are [image_size/16, image_size/16] instead of [image_size/4, image_size/4] expected here, leading to a shape mismatch here.

Which backbone did you use in the paper? I wasn't able to find these architectural details in the main paper or supplement.

Thanks!
Hallee

@j-Halloran
Copy link

yes seconded, either pre-trained weights or a better description of training data would be much appreciated

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants