567b87b1b8e69970c676d2c120cf9ecaa481dc19,models/official/detection/dataloader/classification_parser.py,Parser,_parse_train_data,#Parser#Any#,89

Before Change

    // Resizes image.
    image = tf.image.resize_images(
        cropped_image, self._output_size, method=tf.image.ResizeMethod.BILINEAR)
    image.set_shape([self._output_size[0], self._output_size[1], 3])

    // If bfloat16 is used, casts input image to tf.bfloat16.
    if self._use_bfloat16:
      image = tf.cast(image, dtype=tf.bfloat16)

After Change

    if self._aug_rand_hflip:
      image = tf.image.random_flip_left_right(image)

    image = tf.reshape(image, [self._output_size[0], self._output_size[1], 3])

    // If bfloat16 is used, casts input image to tf.bfloat16.
    if self._use_bfloat16:
      image = tf.cast(image, dtype=tf.bfloat16)

Instances