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

Before Change

    offset = image_info[3, :]
    boxes = input_utils.resize_and_crop_boxes(
        boxes, image_scale, (image_height, image_width), offset)
    if self._include_mask:
      masks = input_utils.resize_and_crop_masks(
          tf.expand_dims(masks, axis=-1),
          image_scale, (image_height, image_width), offset)
      masks = tf.squeeze(masks, axis=-1)

    // Filters out ground truth boxes that are all zeros.
    indices = input_utils.get_non_empty_box_indices(boxes)
    boxes = tf.gather(boxes, indices)
    classes = tf.gather(classes, indices)
    if self._include_mask:

After Change

    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(
          tf.expand_dims(masks, axis=-1),
          cropped_boxes,

Instances