提交 f9c9b163 编写于 作者: J jiangjiajun

modify transforms

上级 d2729260
...@@ -23,7 +23,6 @@ import random ...@@ -23,7 +23,6 @@ import random
import platform import platform
import chardet import chardet
import paddlex.utils.logging as logging import paddlex.utils.logging as logging
from paddlex.cv.transforms.template import TemplateTransforms
class EndSignal(): class EndSignal():
...@@ -233,8 +232,6 @@ class Dataset: ...@@ -233,8 +232,6 @@ class Dataset:
if transforms is None: if transforms is None:
raise Exception("transform should be defined.") raise Exception("transform should be defined.")
self.transforms = transforms self.transforms = transforms
if isinstance(transforms, TemplateTransforms):
self.transforms = transforms.transforms
self.num_workers = num_workers self.num_workers = num_workers
self.buffer_size = buffer_size self.buffer_size = buffer_size
self.parallel_method = parallel_method self.parallel_method = parallel_method
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册