瀏覽代碼

fix imgaug for classification

jiangjiajun 5 年之前
父節點
當前提交
0f1bca2e52
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/transforms/cls_transforms.py

+ 1 - 1
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)