提交 fd2cb4a6 编写于 作者: J jiangjiajun

fix add augmenters

上级 29d7de56
...@@ -96,7 +96,7 @@ class Compose(ClsTransform): ...@@ -96,7 +96,7 @@ class Compose(ClsTransform):
if not isinstance(augmenters, list): if not isinstance(augmenters, list):
raise Exception( raise Exception(
"augmenters should be list type in func add_augmenters()") "augmenters should be list type in func add_augmenters()")
self.transforms = augmenters + self.transforms.transforms self.transforms = augmenters + self.transforms
class RandomCrop(ClsTransform): class RandomCrop(ClsTransform):
......
...@@ -156,7 +156,7 @@ class Compose(DetTransform): ...@@ -156,7 +156,7 @@ class Compose(DetTransform):
if not isinstance(augmenters, list): if not isinstance(augmenters, list):
raise Exception( raise Exception(
"augmenters should be list type in func add_augmenters()") "augmenters should be list type in func add_augmenters()")
self.transforms = augmenters + self.transforms.transforms self.transforms = augmenters + self.transforms
class ResizeByShort(DetTransform): class ResizeByShort(DetTransform):
......
...@@ -112,7 +112,7 @@ class Compose(SegTransform): ...@@ -112,7 +112,7 @@ class Compose(SegTransform):
if not isinstance(augmenters, list): if not isinstance(augmenters, list):
raise Exception( raise Exception(
"augmenters should be list type in func add_augmenters()") "augmenters should be list type in func add_augmenters()")
self.transforms = augmenters + self.transforms.transforms self.transforms = augmenters + self.transforms
class RandomHorizontalFlip(SegTransform): class RandomHorizontalFlip(SegTransform):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册