Skip to content
Snippets Groups Projects
Unverified Commit 85a83f21 authored by Wang Xinjiang's avatar Wang Xinjiang Committed by GitHub
Browse files

prevent possible bugs in loading and transforms caused by shallow copy (#2967)

parent 2fcac966
No related branches found
No related tags found
No related merge requests found
......@@ -186,17 +186,17 @@ class LoadAnnotations(object):
def _load_bboxes(self, results):
ann_info = results['ann_info']
results['gt_bboxes'] = ann_info['bboxes']
results['gt_bboxes'] = ann_info['bboxes'].copy()
gt_bboxes_ignore = ann_info.get('bboxes_ignore', None)
if gt_bboxes_ignore is not None:
results['gt_bboxes_ignore'] = gt_bboxes_ignore
results['gt_bboxes_ignore'] = gt_bboxes_ignore.copy()
results['bbox_fields'].append('gt_bboxes_ignore')
results['bbox_fields'].append('gt_bboxes')
return results
def _load_labels(self, results):
results['gt_labels'] = results['ann_info']['labels']
results['gt_labels'] = results['ann_info']['labels'].copy()
return results
def _poly2mask(self, mask_ann, img_h, img_w):
......
......@@ -603,7 +603,8 @@ class Expand(object):
results['img'] = expand_img
# expand bboxes
for key in results.get('bbox_fields', []):
results[key] += np.tile((left, top), 2).astype(results[key].dtype)
results[key] = results[key] + np.tile(
(left, top), 2).astype(results[key].dtype)
# expand masks
for key in results.get('mask_fields', []):
......@@ -713,7 +714,7 @@ class MinIoURandomCrop(object):
if not mask.any():
continue
for key in results.get('bbox_fields', []):
boxes = results[key]
boxes = results[key].copy()
mask = is_center_of_bboxes_in_patch(boxes, patch)
boxes = boxes[mask]
boxes[:, 2:] = boxes[:, 2:].clip(max=patch[2:])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment