fix: resample classification for dense matches filling

CNES · Oct 11, 2024 · 13b291f · 13b291f
1 parent 05e5de3
commit 13b291f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cars/pipelines/sensor_to_dense_dsm/sensor_to_dense_dsm_pipeline.py b/cars/pipelines/sensor_to_dense_dsm/sensor_to_dense_dsm_pipeline.py
@@ -673,6 +673,7 @@ def run(self):  # noqa C901
                 # with any order. the .1 will be performed before the .2
                 pairs[pair_key]["holes_classif"] = []
                 pairs[pair_key]["holes_poly_margin"] = 0
+                add_classif = False
                 if self.dense_matches_filling_1.used_method == "plane":
                     pairs[pair_key][
                         "holes_classif"
@@ -681,6 +682,7 @@ def run(self):  # noqa C901
                         pairs[pair_key]["holes_poly_margin"],
                         self.dense_matches_filling_1.get_poly_margin(),
                     )
+                    add_classif = True
                 if self.dense_matches_filling_2.used_method == "plane":
                     pairs[pair_key][
                         "holes_classif"
@@ -689,6 +691,7 @@ def run(self):  # noqa C901
                         pairs[pair_key]["holes_poly_margin"],
                         self.dense_matches_filling_2.get_poly_margin(),
                     )
+                    add_classif = True
 
                 pairs[pair_key]["holes_bbox_left"] = []
                 pairs[pair_key]["holes_bbox_right"] = []
@@ -717,7 +720,7 @@ def run(self):  # noqa C901
                         tile_width=None,
                         tile_height=None,
                         add_color=False,
-                        add_classif=False,
+                        add_classif=add_classif,
                     )
 
                     # Generate the holes polygons in epipolar images