decode JPEG image data to numpy array

lte/image_decode.py

@@ -0,0 +1,22 @@
+"""
+Decodes images from JPEG bytes to numpy array.
+"""
+
+import io
+
+from PIL import Image
+import numpy as np
+
+
+def decode(data: "io.BytesIO | bytes") -> np.ndarray:
+    """
+    Decodes a JPEG encoded image and returns it as a numpy array.
+
+    Args:
+        data: bytes object containing the JPEG encoded image
+    Returns:
+        NDArray with in RGB format (Height, Width, 3)
+    """
+    image = Image.open(data, formats=['JPEG'])
+
+    return np.asarray(image)

requirements.txt

@@ -8,6 +8,9 @@ pytest
 opencv-python
 numpy
 
+# Module lte
+Pillow
+
 # Module mavlink
 dronekit