Commit 3cee4c9b authored by Daniel Scheffler's avatar Daniel Scheffler
Browse files
parent 153dca9c
Pipeline #4124 failed with stage
in 1 minute and 48 seconds
......@@ -422,11 +422,11 @@ class kNN_SAM_Classifier(SAM_Classifier):
def label_unclassified_pixels(self, label_unclassified, threshold):
# type: (int, Union[str, int, float]) -> GeoArray
cmap_labelled0 = self._label_unclassified_pixels(
self.cmap.get_subset(zslice=slice(None, 1)), label_unclassified, threshold, self.angles_deg[:, :, 0]
GeoArray(self.cmap[:, :, 0], nodata=self.cmap.nodata), label_unclassified, threshold, self.angles_deg[:, :, 0]
)
cmap = self.cmap
if label_unclassified in cmap_labelled0:
cmap[cmap_labelled0 == label_unclassified] = label_unclassified
if label_unclassified in cmap_labelled0[:]:
cmap[cmap_labelled0[:] == label_unclassified] = label_unclassified
return cmap
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment