facebookresearch / fastMRI

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

Updates for cropping for U-Net and VarNet #56

Closed mmuckley closed 4 years ago

mmuckley commented 4 years ago

This PR updates cropping routine for the U-Net and VarNet. Previously, both of these training modules had a resolution parameter that specified the cropping for the final image. This behavior has been removed and replaced with new logic that reads the final cropped image size from the ismrmrd header. This is particularly applicable to the brain data, where a single resolution parameter does not work for most cases.

The PR also includes minor updates to docstrings.