a30c31849a1ce6791ad201f09507c2f2dfe6db27,tensorflow_datasets/core/features/image_feature.py,Image,encode_sample,#Image#Any#,103

Before Change

      raise ValueError("Could not convert {} to image".format(image_or_path))

    // Check that the image is valid
    if np_image.dtype != np.uint8:
      raise ValueError("Image should be uint8. Detected: {}".format(
          np_image.dtype))
    utils.assert_shape_match(np_image.shape, self._shape)

    // TODO(epot): Should support additional format
    image_format = "png"

After Change

    if isinstance(image_or_path, np.ndarray):
      encoded_image = self._encode_image(image_or_path)
    else:
      with tf.gfile.Open(image_or_path, "rb") as image_f:
        encoded_image = image_f.read()
    return encoded_image

Instances