_strict_random_crop_image: fix mask slice

22ca3a44 · akssri · GitHub · 56765e5b · 22ca3a44
Commit 22ca3a44 authored Jun 27, 2017 by akssri Committed by GitHub Jun 27, 2017
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

object_detection/core/preprocessor.py object_detection/core/preprocessor.py +2 -2

No files found.
--- a/object_detection/core/preprocessor.py
+++ b/object_detection/core/preprocessor.py
@@ -707,8 +707,8 @@ def _strict_random_crop_image(image,
      masks_of_boxes_inside_window = tf.gather(masks, inside_window_ids)
      masks_of_boxes_completely_inside_window = tf.gather(
          masks_of_boxes_inside_window, keep_ids)
-      masks_box_begin = [im_box_begin[2], im_box_begin[0], im_box_begin[1]]
-      masks_box_size = [im_box_size[2], im_box_size[0], im_box_size[1]]
+      masks_box_begin = [0, im_box_begin[0], im_box_begin[1]]
+      masks_box_size = [-1, im_box_size[0], im_box_size[1]]
      new_masks = tf.slice(
          masks_of_boxes_completely_inside_window,
          masks_box_begin, masks_box_size)