Commit afabda6d authored by Jonathan Huang's avatar Jonathan Huang Committed by GitHub
Browse files

Merge pull request #1775 from akssri/patch-1

_strict_random_crop_image: fix mask slice
parents 64fe1dfe 22ca3a44
......@@ -706,8 +706,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)
......
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