5b25005c58bf9aafbdb781b7370825e826b2a237,official/vision/detection/dataloader/maskrcnn_parser.py,Parser,_parse_train_data,#Parser#Any#,145

Before Change

    classes = tf.gather(classes, indices)
    if self._include_mask:
      masks = tf.gather(masks, indices)
      cropped_boxes = boxes + tf.cast(
          tf.tile(tf.expand_dims(offset, axis=0), [1, 2]), dtype=tf.float32)
      cropped_boxes = box_utils.normalize_boxes(
          cropped_boxes, image_info[1, :])
      num_masks = tf.shape(masks)[0]
      masks = tf.image.crop_and_resize(

After Change

      masks = tf.gather(masks, indices)
      // Transfer boxes to the original image space and do normalization.
      cropped_boxes = boxes + tf.tile(tf.expand_dims(offset, axis=0), [1, 2])
      cropped_boxes /= tf.tile(tf.expand_dims(image_scale, axis=0), [1, 2])
      cropped_boxes = box_utils.normalize_boxes(cropped_boxes, image_shape)
      num_masks = tf.shape(masks)[0]
      masks = tf.image.crop_and_resize(
          tf.expand_dims(masks, axis=-1),
          cropped_boxes,

Instances