From fd2cb4a623c2adcf700a634fc6b899697d6404ed Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Mon, 1 Jun 2020 02:06:53 +0000 Subject: [PATCH] fix add augmenters --- paddlex/cv/transforms/cls_transforms.py | 2 +- paddlex/cv/transforms/det_transforms.py | 2 +- paddlex/cv/transforms/seg_transforms.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py index 6dc4ea7..3b2ff78 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 19db331..841a980 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 516d5fb..34ba681 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): -- GitLab