提交 57935dc4 编写于 作者: C cuicheng01

fix img process bug

上级 99527c02
...@@ -29,6 +29,7 @@ from ppcls.data.dataloader.icartoon_dataset import ICartoonDataset ...@@ -29,6 +29,7 @@ from ppcls.data.dataloader.icartoon_dataset import ICartoonDataset
# sampler # sampler
from ppcls.data.dataloader.DistributedRandomIdentitySampler import DistributedRandomIdentitySampler from ppcls.data.dataloader.DistributedRandomIdentitySampler import DistributedRandomIdentitySampler
from ppcls.data import preprocess
from ppcls.data.preprocess import transform from ppcls.data.preprocess import transform
...@@ -46,7 +47,7 @@ def create_operators(params): ...@@ -46,7 +47,7 @@ def create_operators(params):
dict) and len(operator) == 1, "yaml format error" dict) and len(operator) == 1, "yaml format error"
op_name = list(operator)[0] op_name = list(operator)[0]
param = {} if operator[op_name] is None else operator[op_name] param = {} if operator[op_name] is None else operator[op_name]
op = getattr(imaug, op_name)(**param) op = getattr(preprocess, op_name)(**param)
ops.append(op) ops.append(op)
return ops return ops
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册