|
@@ -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)
|