Hugging Face's logo

VBoussot
/
ImpactReg

Image-to-Image
konfai
medical-imaging
registration
deformable
multimodal
ct
mri
cbct
Model card Files
xet
 Community
ImpactReg
File size: 3,313 Bytes
6afc1ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
Predictor:
  Model:
    classpath: Model:RegistrationNet
    RegistrationNet:
      scales:
      - 4
      - 2
      - 1
      affine_iterations:
      - 200
      - 100
      - 50
      deformable_iterations:
      - 200
      - 100
      - 50
      cc_kernel: 5
      affine_metric: mi
      affine_lr: 0.003
      deformable_method: syn
      deformable_metric: impact
      deformable_lr: 0.25
      integrator_n: 10
      smooth_warp_sigma: 0.5
      smooth_grad_sigma: 1.0
      seed: 42
      models:
        '0':
          ref: VBoussot/impact-torchscript-models:TS/M291.pt
          layers_mask: '01'
          layers_weight: 1.0
          pca: 0
          distance: L1
      outputs_criterions: None
  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-FireANTs-IMPACT
  manual_seed: 42
  gpu_checkpoints: None
  images_log: None
  combine: Mean
  autocast: false
  data_log: None