skip copy bboxes in horizontal flip augmentation for speedup (#606)

Megvii-BaseDetection · Aug 28, 2021 · 3571419 · 3571419
1 parent 40bde39
commit 3571419
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/yolox/data/data_augment.py b/yolox/data/data_augment.py
@@ -144,7 +144,6 @@ def _mirror(image, boxes, prob=0.5):
     _, width, _ = image.shape
     if random.random() < prob:
         image = image[:, ::-1]
-        boxes = boxes.copy()
         boxes[:, 0::2] = width - boxes[:, 2::-2]
     return image, boxes