search

StanfordMIMI / DDM2

[ICLR2023] Official repository of DDM2: Self-Supervised Diffusion MRI Denoising with Generative Diffusion Models
125 stars 20 forks source link

Trained models #4

Open prinshul opened 1 year ago

prinshul commented 1 year ago

Hi, Thank you for sharing your code.

Can you please share the trained models (both 2D and 3D)?

tiangexiang commented 1 year ago

Thank you for your interest in our work! Unfortunately, we don't plan to release the trained model, since we have refactored the code quite a lot and the trained model cannot be loaded directly in the current repo due to inconsistent names/structures. Note that our method is dataset-specific, such that the trained model on one dataset can not be used to denoise any other datasets. However, we do provide the denoised data for all the four datasets we presented in the paper: https://figshare.com/s/6275f40c32f67e3b6083

prinshul commented 1 year ago

Sure. I will train the model from scratch. Waiting for the metric calculation script on test data for Hardi dataset. Thanks.