Revert "Fix BGR->RGB Bug in albumentations ultralytics#8641" (ultraly…

…tics#8727) Revert "Fix BGR->RGB Bug in albumentations ultralytics#8641 (ultralytics#8695)" This reverts commit 2e1291f.
ctjanuhowski · Sep 8, 2022 · 467f18e · 467f18e
1 parent a6ae28f
commit 467f18e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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