search

yang-song / score_inverse_problems

Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"
225 stars 25 forks source link

Missing source files ? #3

Closed zaccharieramzi closed 2 years ago

zaccharieramzi commented 2 years ago

Hi,

I am very interested in your work that I find excellent, and I would like to reproduce some results as well as extend it a bit. I am having a hard time just running the main.py script. In the datasets.py file, many files/packages are imported among which ct2d or fastmri_knee_single. I can't seem to find these files/folders anywhere in the repository, maybe they are missing, or their names have changed. I am investigating the latter case, but in the meantime if it's just a matter of missing files, could you add them?

Thanks a lot, Cheers

yang-song commented 2 years ago

ct2d and fastmri_knee_single are all preprocessed dataset files. They are huge and we are right now figuring out the best way to upload them. That said, the code itself should be quite general and should work for any other datasets.

zaccharieramzi commented 2 years ago

They are huge python files? I am asking because they seem to be imported in the datasets.py file.

yang-song commented 2 years ago

For now, you can just remove those imports as they are not used in our experiments. I will also clean up the code later. We will upload datasets for ct2d_320, ldct_512 , and brats

zaccharieramzi commented 2 years ago

Ok makes sense, closing then

hosseinaskari-cs commented 1 year ago

ct2d and fastmri_knee_single are all preprocessed dataset files. They are huge and we are right now figuring out the best way to upload them. That said, the code itself should be quite general and should work for any other datasets.

At least please upload the code for pre-processing the datasets.