Sfoglia il codice sorgente

fix add augmenters

jiangjiajun 5 anni fa
parent
commit
fd2cb4a623

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

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

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