search

laura-rieger / deep-explanation-penalization

Code for using CDEP from the paper "Interpretations are useful: penalizing explanations to align neural networks with prior knowledge" https://arxiv.org/abs/1909.13584
MIT License
127 stars 14 forks source link

How do I get segmentation files of ISIC dataset? #5

Closed Juna2 closed 4 years ago

Juna2 commented 4 years ago

Hi, I appreciate your awesome code!! Btw how do I get segmentation files of the ISIC dataset? I can't find any code in 02_sort_imgs.py that save segmentation files in "segmentation_path = os.path.join(data_path, "segmentation")"

I found in another issue that I can download SONIC study in ISIC website but I still don't know if this data is already included in the dataset I can download with 01_download_imgs.py file.

So which data of the study in the ISIC dataset did you exactly use for the ISIC result in your paper? There were the following studies.

2018 JID Editorial Images (0 / 100) Dermoscopedia (CC-BY) (0 / 5) HAM10000 (0 / 10015) MSK-1 (0 / 1100) MSK-2 (0 / 1535) MSK-3 (0 / 225) MSK-4 (0 / 947) MSK-5 (0 / 111) SONIC (9251) UDA-1 (0 / 557) UDA-2 (0 / 60)

laura-rieger commented 4 years ago

Hi Jeon, glad you like the work! The segmentations need to be downloaded manually. I put them in a Google Drive here I am not entirely sure I understand your second question. We used all images in the ISIC dataset (so including the SONIC dataset) barring images that did not have a conclusive diagnosis. Does this answer your question? Best, Laura

Juna2 commented 4 years ago

OMG sorry I didn't know that the link was in readme.md!!! Thank you so much for such a quick reply. I appreciate that.

laura-rieger commented 4 years ago

No worries, I just added the link after you pointed out that it's not currently there, so that was on me.