diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py index 6dc4ea7b95d876ae896c77395ab155bec1727a8a..3b2ff789f280d1006c55bf8a0ed8ab4e5ebd5967 100644 --- a/paddlex/cv/transforms/cls_transforms.py +++ b/paddlex/cv/transforms/cls_transforms.py @@ -96,7 +96,7 @@ class Compose(ClsTransform): if not isinstance(augmenters, list): raise Exception( "augmenters should be list type in func add_augmenters()") - self.transforms = augmenters + self.transforms.transforms + self.transforms = augmenters + self.transforms class RandomCrop(ClsTransform): diff --git a/paddlex/cv/transforms/det_transforms.py b/paddlex/cv/transforms/det_transforms.py index 19db33173b87b7cc20b87054cfbc1241176abc58..841a980ecd621cc8d1ac494f41e8bebf1d40c8f7 100644 --- a/paddlex/cv/transforms/det_transforms.py +++ b/paddlex/cv/transforms/det_transforms.py @@ -156,7 +156,7 @@ class Compose(DetTransform): if not isinstance(augmenters, list): raise Exception( "augmenters should be list type in func add_augmenters()") - self.transforms = augmenters + self.transforms.transforms + self.transforms = augmenters + self.transforms class ResizeByShort(DetTransform): diff --git a/paddlex/cv/transforms/seg_transforms.py b/paddlex/cv/transforms/seg_transforms.py index 516d5fb14fddee90e38b716ccaf4091558c300a8..34ba681b5be197549ce680e8bd762e2cab06f00b 100644 --- a/paddlex/cv/transforms/seg_transforms.py +++ b/paddlex/cv/transforms/seg_transforms.py @@ -112,7 +112,7 @@ class Compose(SegTransform): if not isinstance(augmenters, list): raise Exception( "augmenters should be list type in func add_augmenters()") - self.transforms = augmenters + self.transforms.transforms + self.transforms = augmenters + self.transforms class RandomHorizontalFlip(SegTransform):