Skip to content
Snippets Groups Projects
Commit b1034740 authored by Yanbing Dong's avatar Yanbing Dong
Browse files

Mod: modify the range of bbox in BboxTransform

parent 70383d46
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmdet/datasets/transforms.py
+ 2
− 2
View file @ b1034740
...@@ -76,8 +76,8 @@ class BboxTransform(object): ...@@ -76,8 +76,8 @@ class BboxTransform(object):
gt_bboxes = bboxes * scale_factor gt_bboxes = bboxes * scale_factor
if flip: if flip:
gt_bboxes = bbox_flip(gt_bboxes, img_shape) gt_bboxes = bbox_flip(gt_bboxes, img_shape)
gt_bboxes[:, 0::2] = np.clip(gt_bboxes[:, 0::2], 0, img_shape[1]) gt_bboxes[:, 0::2] = np.clip(gt_bboxes[:, 0::2], 0, img_shape[1] - 1)
gt_bboxes[:, 1::2] = np.clip(gt_bboxes[:, 1::2], 0, img_shape[0]) gt_bboxes[:, 1::2] = np.clip(gt_bboxes[:, 1::2], 0, img_shape[0] - 1)
if self.max_num_gts is None: if self.max_num_gts is None:
return gt_bboxes return gt_bboxes
else: else:
......
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