添加方向分类器导出支持

This commit is contained in:
WenmuZhou 2020-09-16 10:43:11 +08:00
parent f3f2b38efb
commit 25ff7eb517
1 changed files with 2 additions and 1 deletions

View File

@ -79,6 +79,7 @@ class ClsModel(object):
outputs = {'total_loss': loss, 'decoded_out': \
predicts['decoded_out'], 'label': label, 'acc': acc}
return loader, outputs
elif mode == "export":
return [image, predicts]
else:
return loader, predicts