소스 검색

fix add augmenters

jiangjiajun 5 년 전
부모
커밋
fd2cb4a623
3개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      paddlex/cv/transforms/cls_transforms.py
  2. 1 1
      paddlex/cv/transforms/det_transforms.py
  3. 1 1
      paddlex/cv/transforms/seg_transforms.py

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