浏览代码

modify transforms

jiangjiajun 5 年之前
父节点
当前提交
f9c9b1630a
共有 1 个文件被更改,包括 0 次插入3 次删除
  1. 0 3
      paddlex/cv/datasets/dataset.py

+ 0 - 3
paddlex/cv/datasets/dataset.py

@@ -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:
         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