Skip to content

GitLab

Commit 8a120d0e authored by syiming's avatar syiming
Browse files

change crop_and_resize_fn input to make it consistant with multilevel function

parent b48046ed
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @ 8a120d0e
...@@ -2549,8 +2549,8 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -2549,8 +2549,8 @@ class FasterRCNNMetaArch(model.DetectionModel):
image_shape[1], image_shape[2], check_range=False).get() image_shape[1], image_shape[2], check_range=False).get()
flat_cropped_gt_mask = self._crop_and_resize_fn( flat_cropped_gt_mask = self._crop_and_resize_fn(
tf.expand_dims(flat_gt_masks, -1), [tf.expand_dims(flat_gt_masks, -1)],
tf.expand_dims(flat_normalized_proposals, axis=1), tf.expand_dims(flat_normalized_proposals, axis=1), None,
[mask_height, mask_width]) [mask_height, mask_width])
# Without stopping gradients into cropped groundtruth masks the # Without stopping gradients into cropped groundtruth masks the
# performance with 100-padded groundtruth masks when batch size > 1 is # performance with 100-padded groundtruth masks when batch size > 1 is
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment