Went to uint8

WorldCereal · Jun 20, 2024 · c14db91 · c14db91
1 parent 1166fa0
commit c14db91
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/inference/cropland_mapping.py b/scripts/inference/cropland_mapping.py
@@ -8,7 +8,7 @@
 from openeo_gfmap.backend import Backend, BackendContext
 from openeo_gfmap.features.feature_extractor import apply_feature_extractor
 from openeo_gfmap.inference.model_inference import apply_model_inference
-from openeo_gfmap.preprocessing.scaling import compress_uint16
+from openeo_gfmap.preprocessing.scaling import compress_uint8
 
 from worldcereal.openeo.feature_extractor import PrestoFeatureExtractor
 from worldcereal.openeo.inference import CroplandClassifier
@@ -106,8 +106,8 @@
         ],
     )
 
-    # Cast to uint16
-    classes = compress_uint16(classes)
+    # Cast to uint8
+    classes = compress_uint8(classes)
 
     classes.execute_batch(
         outputfile=args.output_path,