facebookresearch / fastMRI

A large-scale dataset of both raw MRI measurements and clinical MRI images.
https://fastmri.org
MIT License
1.29k stars 370 forks source link

Add fastMRI+ data compatibility to SliceDataset #178

Closed mmuckley closed 2 years ago

mmuckley commented 2 years ago

We would like to incorporate functionality for using the fastMRI+ dataset in this GitHub repository. In short, this should:

All of this should be feasible by subclassing and extending SliceDataset in mri_data.py. I'm posting the issue for anyone that might want to pick up the project.

mmuckley commented 2 years ago

CC @hansenms

mattnsoleil commented 2 years ago

Hello,

In VarNet, The dataset just loads images with the PIL library and does not convert them into necessary types like list or Tensor in SliceDataset. Consequently, this gives an error like TypeError: batch must contain tensors, numbers, dicts or lists; found object. Is it a solution or an update for this problem?

mmuckley commented 2 years ago

@mattnsoleil that is not related to implementing fastMRI+. Please open as separate issue.

Gaskell-1206 commented 2 years ago

Hi everyone, I'm Shengjia or Gaskell. I'm working on this issue. If you have any questions or other concerns, please feel free to talk to me. Thanks!