Skip to content

GitLab

"torchvision/csrc/vscode:/vscode.git/clone" did not exist on "8520f0bea4a09e60a217fe3a8cf24b8f733ec16c"
Commit b1034740 authored by Yanbing Dong's avatar Yanbing Dong
Browse files

Mod: modify the range of bbox in BboxTransform

parent 70383d46
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmdet/datasets/transforms.py
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:
......
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