search

ishaanb92 / GeneralizedProbabilisticUNet

PyTorch model for the Generalized Probabilistic U-Net. For more details see: https://www.melba-journal.org/papers/2023:005.html
MIT License
13 stars 7 forks source link

Question about data split #1

Closed cviviers closed 7 months ago

cviviers commented 2 years ago

Hi Ishaan, Very interesting work and result. I was wondering if you had split the data into train/val/test on a per patient level? It was not clear to me from your paper.

Thanks in advance for the feedback.

ishaanb92 commented 2 years ago

Hi Christiaan,

Thanks for your interest! Apologies for the delay in the reply, I was away.

I used a pre-processed version of the LIDC-IDRI dataset available at https://drive.google.com/drive/folders/1xKfKCQo8qa6SAr3u7qWNtQjIphIrvmd5

To answer your question, I've added the script I used to split the dataset under the 'data' folder. Hope that helps, and let me know if you have further questions.

Cheers, Ishaan