fix bug in creating preprocessing transform object
@@ -33,8 +33,9 @@ def get_postprocess(postprocesses, cfg, compose=True):
def create_dataset(framework, data_source, cfg_preprocess):
transform_list = []
# generate framework specific transforms
+ preprocess = None
+ if cfg_preprocess is not None:
preprocesses = TRANSFORMS(framework, 'preprocess')
-
preprocess = get_preprocess(preprocesses, cfg_preprocess)
# even we can unify transform, how can we handle the IO, or we do the transform here
datasets = DATASETS(framework)
