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

Before Change

  def encode_sample(self, image_or_path):
    Convert the given image into a dict convertible to tf example.
    if isinstance(image_or_path, six.string_types):
      // TODO(epot): np_image = load_image_from_disk(image_or_path)
      raise NotImplementedError
    elif isinstance(image_or_path, np.ndarray):
      np_image = image_or_path
    else:
      // Could also add PIL support
      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)

After Change

      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

  def decode_sample(self, sample):
    Reconstruct the image from the tf example.

Instances