Unverified 提交 2e1291fd authored 作者: UnglvKitDe's avatar UnglvKitDe 提交者: GitHub

Fix BGR->RGB Bug in albumentations #8641 (#8695)

* Fix BGR->RGB Bug in albumentations https://github.com/ultralytics/yolov5/issues/8641 * Change transform methode from cv2 to numpy * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Simplify * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update augmentations.py Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 b3678601
...@@ -39,8 +39,9 @@ class Albumentations: ...@@ -39,8 +39,9 @@ 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, bboxes=labels[:, 1:], class_labels=labels[:, 0]) # transformed new = self.transform(image=im[..., ::-1], 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'])]) im = new['image'][..., ::-1] # RGB to BGR
labels = np.array([[c, *b] for c, b in zip(new['class_labels'], new['bboxes'])])
return im, labels return im, labels
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论