Fix PIL image conversion for uint16 multi-channel arrays.

PiperOrigin-RevId: 911782033

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/utils/image_utils.py

@@ -177,16 +177,19 @@ def create_thumbnail(
   if use_colormap:  # Apply the colormap first as it modify the shape/dtype
     ex = apply_colormap(ex)
 
-  _, _, c = ex.shape
+  # PIL.Image.fromarray doesn't support uint16 for >1 channels.
+  # https://github.com/python-pillow/Pillow/blob/11.0.0/src/PIL/Image.py#L3225
+  if ex.dtype == np.uint16:
+    ex = (ex / 257).astype(np.uint8)
+
+  c = ex.shape[-1] if ex.ndim == 3 else 1
   postprocess = _postprocess_noop
   if c == 1:
     ex = ex.squeeze(axis=-1)
     mode = 'L'
-  elif ex.dtype == np.uint16:
-    mode = 'I;16'
-    postprocess = _postprocess_convert_rgb
   else:
     mode = None
+    postprocess = _postprocess_convert_rgb
   img = PIL_Image.fromarray(ex, mode=mode)
   img = postprocess(img)
   if default_dimensions: