"Elements in transforms should be defined in 'paddlex.cls.transforms' or class of imgaug.augmenters.Augmenter, see docs here: https://paddlex.readthedocs.io/zh_CN/latest/apis/transforms/"
"Elements in transforms should be defined in 'paddlex.det.transforms' or class of imgaug.augmenters.Augmenter, see docs here: https://paddlex.readthedocs.io/zh_CN/latest/apis/transforms/"
# See the License for the specific language governing permissions and
# limitations under the License.
importos
from.opsimport*
from.imgaug_supportimportexecute_imgaug
importrandom
...
...
@@ -45,7 +46,7 @@ class Compose(SegTransform):
"""
def__init__(self,transforms):
def__init__(self,transforms,vdl_save_dir=None):
ifnotisinstance(transforms,list):
raiseTypeError('The transforms must be a list!')
iflen(transforms)<1:
...
...
@@ -61,8 +62,24 @@ class Compose(SegTransform):
raiseException(
"Elements in transforms should be defined in 'paddlex.seg.transforms' or class of imgaug.augmenters.Augmenter, see docs here: https://paddlex.readthedocs.io/zh_CN/latest/apis/transforms/"