提交 0f1bca2e 编写于 作者: J jiangjiajun

fix imgaug for classification

上级 9165743f
...@@ -86,7 +86,7 @@ class Compose(ClsTransform): ...@@ -86,7 +86,7 @@ class Compose(ClsTransform):
else: else:
import imgaug.augmenters as iaa import imgaug.augmenters as iaa
if isinstance(op, iaa.Augmenter): if isinstance(op, iaa.Augmenter):
im, = execute_imgaug(op, im) im = execute_imgaug(op, im)
outputs = (im, ) outputs = (im, )
if label is not None: if label is not None:
outputs = (im, label) outputs = (im, label)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册