From 0f1bca2e52ad32368fc1a0505004eebca42be770 Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Mon, 18 May 2020 21:31:23 +0800 Subject: [PATCH] fix imgaug for classification --- paddlex/cv/transforms/cls_transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py index 499baf7..844516c 100644 --- a/paddlex/cv/transforms/cls_transforms.py +++ b/paddlex/cv/transforms/cls_transforms.py @@ -86,7 +86,7 @@ class Compose(ClsTransform): else: import imgaug.augmenters as iaa if isinstance(op, iaa.Augmenter): - im, = execute_imgaug(op, im) + im = execute_imgaug(op, im) outputs = (im, ) if label is not None: outputs = (im, label) -- GitLab