.

sorghum_pipeline/data/preprocessor.py

@@ -27,7 +27,10 @@ class ImagePreprocessor:
         boxes = [(j, i, j + d, i + d)
                  for i, j in product(range(0, pil_img.height, d),
                                     range(0, pil_img.width,  d))]
-        stack = np.stack([np.array(pil_img.crop(b), float) for b in boxes], axis=-1)
+        # Convert each quadrant to grayscale so each band is single-channel
+        stack = np.stack([
+            np.array(pil_img.crop(b).convert('L'), dtype=float)
+        for b in boxes], axis=-1)
         # bands come in order: [green, red, red_edge, nir]
         green, red, red_edge, nir = np.split(stack, 4, axis=-1)
         # build composite using bands: Red, Red-Edge, Green