From 125bc2a87832f3285c4359a0df107ade98412d65 Mon Sep 17 00:00:00 2001
From: sunyanfang01 <sunyanfang01@baidu.com>
Date: Thu, 4 Jun 2020 10:42:42 +0800
Subject: [PATCH] fix the seg transf

---
 paddlex/cv/transforms/seg_transforms.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddlex/cv/transforms/seg_transforms.py b/paddlex/cv/transforms/seg_transforms.py
index d3c6764..a17f97b 100644
--- a/paddlex/cv/transforms/seg_transforms.py
+++ b/paddlex/cv/transforms/seg_transforms.py
@@ -44,13 +44,14 @@ class Compose(SegTransform):
 
     """
 
-    def __init__(self, transforms):
+    def __init__(self, transforms, batch_transforms=None):
         if not isinstance(transforms, list):
             raise TypeError('The transforms must be a list!')
         if len(transforms) < 1:
             raise ValueError('The length of transforms ' + \
                             'must be equal or larger than 1!')
         self.transforms = transforms
+        self.batch_transforms = batch_transforms
         self.to_rgb = False
         # 检查transforms里面的操作，目前支持PaddleX定义的或者是imgaug操作
         for op in self.transforms:
-- 
GitLab