ImpactReg / Generic_Rigid /Prediction.yml
Valentin Boussot
refactor: split elastix engine and type model parameters across presets
269b189
Raw
History Blame Contribute Delete
2.86 kB
Predictor:
Model:
classpath: Model:RegistrationNet
RegistrationNet:
engine: elastix
parameter_maps:
- Parameters_Rigid.txt
outputs_criterions: None
max_iterations: 0
spatial_samples: 2048
parameter_overrides: []
Dataset:
groups_src:
Volume_0:
groups_dest:
Fixed:
transforms:
TensorCast:
dtype: float32
inverse: false
patch_transforms: None
is_input: true
Volume_1:
groups_dest:
Moving:
transforms:
TensorCast:
dtype: float32
inverse: false
patch_transforms: None
is_input: true
Volume_2:
groups_dest:
FixedMask:
transforms:
TensorCast:
dtype: float32
inverse: false
patch_transforms: None
is_input: true
Volume_3:
groups_dest:
MovingMask:
transforms:
TensorCast:
dtype: float32
inverse: false
patch_transforms: None
is_input: true
augmentations:
DataAugmentation_0:
data_augmentations:
Flip:
f_prob:
- 0
- 0.5
- 0.5
vector_field: true
prob: 1
nb: 2
Patch:
patch_size: None
overlap: None
mask: None
pad_value: None
extend_slice: 0
subset: None
filter: None
dataset_filenames:
- ./Dataset/:mha
use_cache: false
batch_size: 1
num_workers: None
pin_memory: false
prefetch_factor: None
persistent_workers: None
outputs_dataset:
MovedImage:
OutputDataset:
name_class: OutSameAsGroupDataset
before_reduction_transforms: None
after_reduction_transforms: None
final_transforms:
TensorCast:
dtype: float32
inverse: false
dataset_filename: Moved:mha
group: Moved
same_as_group: Volume_0:Fixed
patch_combine: None
inverse_transform: false
reduction: Mean
Mean: {}
DisplacementField:
OutputDataset:
name_class: OutSameAsGroupDataset
before_reduction_transforms: None
after_reduction_transforms: None
final_transforms:
TensorCast:
dtype: float32
inverse: false
dataset_filename: DVF:mha
group: DVF
same_as_group: Volume_0:Fixed
patch_combine: None
inverse_transform: false
reduction: Mean
Mean: {}
train_name: ImpactReg-Generic-Rigid
manual_seed: 42
gpu_checkpoints: None
images_log: None
combine: Mean
autocast: false
data_log: None