trained_model_assets: save predictions under reliability channel, and…

… only save building and ground probas
IGNF · Oct 19, 2023 · 6c63537 · 6c63537
1 parent aab439c
commit 6c63537
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...odel_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml b/...odel_assets/20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
@@ -243,6 +243,6 @@ predict:
     _target_: myria3d.models.interpolation.Interpolator
     interpolation_k: 10
     classification_dict: ${dataset_description.classification_dict}
-    probas_to_save: all
-    predicted_classification_channel: PredictedClassification
+    probas_to_save: [building,ground]
+    predicted_classification_channel: reliability
     entropy_channel: entropy