Valentin Boussot commited on
Commit ·
39b84cb
1
Parent(s): 0e9eb18
Add finetune models
Browse files- Finetune/Prediction.yml +14 -3
Finetune/Prediction.yml
CHANGED
|
@@ -11,26 +11,37 @@ Predictor:
|
|
| 11 |
MASK:
|
| 12 |
transforms:
|
| 13 |
KonfAIInference:
|
| 14 |
-
repo_id: VBoussot/
|
| 15 |
model_name: body
|
| 16 |
checkpoints_name:
|
| 17 |
-
-
|
| 18 |
number_of_tta: 0
|
| 19 |
number_of_mc_dropout: 0
|
| 20 |
per_channel: false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
Save:
|
| 22 |
dataset: ./Dataset:mha
|
| 23 |
patch_transforms: None
|
| 24 |
is_input: false
|
| 25 |
Volume:
|
| 26 |
transforms:
|
|
|
|
|
|
|
|
|
|
| 27 |
Clip/0:
|
| 28 |
min_value: -1024
|
| 29 |
max_value: 5000
|
| 30 |
save_clip_min: false
|
| 31 |
save_clip_max: false
|
| 32 |
mask: None
|
| 33 |
-
TensorCast:
|
| 34 |
dtype: int16
|
| 35 |
inverse: false
|
| 36 |
ResampleToResolution:
|
|
|
|
| 11 |
MASK:
|
| 12 |
transforms:
|
| 13 |
KonfAIInference:
|
| 14 |
+
repo_id: VBoussot/ImpactSeg
|
| 15 |
model_name: body
|
| 16 |
checkpoints_name:
|
| 17 |
+
- Body.pt
|
| 18 |
number_of_tta: 0
|
| 19 |
number_of_mc_dropout: 0
|
| 20 |
per_channel: false
|
| 21 |
+
ResampleToResolution:
|
| 22 |
+
spacing:
|
| 23 |
+
- 1
|
| 24 |
+
- 1
|
| 25 |
+
- 3
|
| 26 |
+
inverse: false
|
| 27 |
+
Dilate:
|
| 28 |
+
dilate: 5
|
| 29 |
Save:
|
| 30 |
dataset: ./Dataset:mha
|
| 31 |
patch_transforms: None
|
| 32 |
is_input: false
|
| 33 |
Volume:
|
| 34 |
transforms:
|
| 35 |
+
TensorCast/0:
|
| 36 |
+
dtype: float16
|
| 37 |
+
inverse: false
|
| 38 |
Clip/0:
|
| 39 |
min_value: -1024
|
| 40 |
max_value: 5000
|
| 41 |
save_clip_min: false
|
| 42 |
save_clip_max: false
|
| 43 |
mask: None
|
| 44 |
+
TensorCast/1:
|
| 45 |
dtype: int16
|
| 46 |
inverse: false
|
| 47 |
ResampleToResolution:
|