direct-vsharp-multianatomy / vsharp_brain.yaml
georgeyiasemis's picture
Add universal + breast vSHARP checkpoints; refresh READMEs and inference YAMLs
b457626 verified
Raw
History Blame Contribute Delete
1.48 kB
model:
model_name: vsharp.vsharp.VSharpNet
num_steps: 16
num_steps_dc_gd: 12
image_init: SENSE
no_parameter_sharing: true
auxiliary_steps: -1
image_model_architecture: UNET
initializer_channels:
- 32
- 32
- 32
- 64
- 64
initializer_dilations:
- 1
- 1
- 2
- 4
- 8
initializer_multiscale: 1
initializer_activation: PRELU
image_unet_num_filters: 16
image_unet_num_pool_layers: 5
image_unet_dropout: 0.0
image_unet_conv_out_bias: true
additional_models:
sensitivity_model:
model_name: unet.unet_2d.UnetModel2d
in_channels: 2
out_channels: 2
num_filters: 16
num_pool_layers: 5
dropout_probability: 0.0
conv_out_bias: true
physics:
forward_operator: fft2
backward_operator: ifft2
inference:
batch_size: 1
crop: header
dataset:
name: FastMRI
transforms:
use_seed: true
delete_kspace: false
masking:
name: FastMRIRandom
accelerations:
- 4
center_fractions:
- 0.08
cropping:
image_center_crop: false
sensitivity_map_estimation:
estimate_sensitivity_maps: true
normalization:
scaling_key: masked_kspace
scale_percentile: 0.995