Be explicit about PILImage.Image conversion

rerun-io · Dec 13, 2024 · 0d48ea8 · 0d48ea8
1 parent 231e60c
commit 0d48ea8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rerun_py/rerun_sdk/rerun/archetypes/image_ext.py b/rerun_py/rerun_sdk/rerun/archetypes/image_ext.py
@@ -38,9 +38,13 @@
 
 
 def _to_numpy(tensor: ImageLike) -> npt.NDArray[Any]:
+    from PIL import Image as PILImage
+
     # isinstance is 4x faster than catching AttributeError
     if isinstance(tensor, np.ndarray):
         return tensor
+    if isinstance(tensor, PILImage.Image):
+        return np.array(tensor, copy=False)
 
     try:
         # Make available to the cpu