瀏覽代碼

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 chardet
 import paddlex.utils.logging as logging
-from paddlex.cv.transforms.template import TemplateTransforms
 
 
 class EndSignal():
@@ -233,8 +232,6 @@ class Dataset:
         if transforms is None:
             raise Exception("transform should be defined.")
         self.transforms = transforms
-        if isinstance(transforms, TemplateTransforms):
-            self.transforms = transforms.transforms
         self.num_workers = num_workers
         self.buffer_size = buffer_size
         self.parallel_method = parallel_method