b6437e4dd115c5b290eb84b0620610b497293609,models/official/detection/serving/inputs.py,,raw_image_tensor_input,#,59

Before Change

      dtype=tf.float32,
      shape=(batch_size, image_height, image_width, 3))

  image_shape = tf.cast(tf.shape(placeholder)[1:3], dtype=tf.float32)

  image_info_per_image = tf.stack(
      [image_shape,
       image_shape,
       tf.constant([1.0, 1.0], dtype=tf.float32),
       tf.constant([0.0, 0.0], dtype=tf.float32)])
  if batch_size == 1:
    images_info = tf.expand_dims(image_info_per_image, axis=0)
  else:
    images_info = tf.tile(
        tf.expand_dims(image_info_per_image, axis=0), [batch_size, 1, 1])

After Change

      shape=(batch_size, image_height, image_width, 3))

  image_info_per_image = [
      [image_height, image_width],
      [image_height, image_width],
      [1.0, 1.0],
      [0.0, 0.0]]

Instances