提交 25ff7eb5 编写于 作者: W WenmuZhou

添加方向分类器导出支持

上级 f3f2b38e
...@@ -79,6 +79,7 @@ class ClsModel(object): ...@@ -79,6 +79,7 @@ class ClsModel(object):
outputs = {'total_loss': loss, 'decoded_out': \ outputs = {'total_loss': loss, 'decoded_out': \
predicts['decoded_out'], 'label': label, 'acc': acc} predicts['decoded_out'], 'label': label, 'acc': acc}
return loader, outputs return loader, outputs
elif mode == "export":
return [image, predicts]
else: else:
return loader, predicts return loader, predicts
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册