convert from bool to uint8 before writing output (#74)

ubarsc · Mar 11, 2024 · 866ee42 · 866ee42
1 parent c880096
commit 866ee42
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fmask/fmask.py b/fmask/fmask.py
@@ -769,6 +769,8 @@ def cloudFinalPass(info, inputs, outputs, otherargs):
     # they set a pixel to cloud if 5 or more of its 3x3 neighbours is cloud. 
     # This little incantation will do exactly the same. 
     bufferedCloudmask = (uniform_filter(cloudmask * 2.0, size=3) >= 1.0)
+    # convert to int so we can write it with GDAL
+    bufferedCloudmask = bufferedCloudmask.astype(numpy.uint8)
 
     bufferedCloudmask[nullmask] = 0