Fix ISIC dataloader, shape error in UNet & questions about paper #1
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
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 recommendedunet
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