From bb5f9469dfcd10d1bea5ccb093330553c340a55e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 26 Jul 2022 18:46:07 +0200 Subject: [PATCH] Revert "Fix BGR->RGB Bug in albumentations #8641 (#8695)" This reverts commit 2e1291fdce26b3cff213e9e7ee8c196fa263b688. --- utils/augmentations.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/utils/augmentations.py b/utils/augmentations.py index 97506ae25123..3f764c06ae3b 100644 --- a/utils/augmentations.py +++ b/utils/augmentations.py @@ -39,9 +39,8 @@ def __init__(self): def __call__(self, im, labels, p=1.0): if self.transform and random.random() < p: - new = self.transform(image=im[..., ::-1], bboxes=labels[:, 1:], class_labels=labels[:, 0]) # transformed - im = new['image'][..., ::-1] # RGB to BGR - labels = np.array([[c, *b] for c, b in zip(new['class_labels'], new['bboxes'])]) + new = self.transform(image=im, bboxes=labels[:, 1:], class_labels=labels[:, 0]) # transformed + im, labels = new['image'], np.array([[c, *b] for c, b in zip(new['class_labels'], new['bboxes'])]) return im, labels