未验证 提交 2b7209c3 编写于 作者: W wangxinxin08 提交者: GitHub

fix num_classes (#2844)

上级 dafdb64f
......@@ -44,9 +44,11 @@ class Compose(object):
for t in self.transforms:
for k, v in t.items():
op_cls = getattr(transform, k)
self.transforms_cls.append(op_cls(**v))
if hasattr(op_cls, 'num_classes'):
op_cls.num_classes = num_classes
f = op_cls(**v)
if hasattr(f, 'num_classes'):
f.num_classes = num_classes
self.transforms_cls.append(f)
def __call__(self, data):
for f in self.transforms_cls:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册