Unverified 提交 c775a296 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Revert "Fix BGR->RGB Bug in albumentations #8641" (#8727)

Revert "Fix BGR->RGB Bug in albumentations #8641 (#8695)" This reverts commit 2e1291fd.
上级 d5116bbe
...@@ -39,9 +39,8 @@ class Albumentations: ...@@ -39,9 +39,8 @@ class Albumentations:
def __call__(self, im, labels, p=1.0): def __call__(self, im, labels, p=1.0):
if self.transform and random.random() < p: if self.transform and random.random() < p:
new = self.transform(image=im[..., ::-1], bboxes=labels[:, 1:], class_labels=labels[:, 0]) # transformed new = self.transform(image=im, bboxes=labels[:, 1:], class_labels=labels[:, 0]) # transformed
im = new['image'][..., ::-1] # RGB to BGR im, labels = new['image'], np.array([[c, *b] for c, b in zip(new['class_labels'], new['bboxes'])])
labels = np.array([[c, *b] for c, b in zip(new['class_labels'], new['bboxes'])])
return im, labels return im, labels
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论