添加方向分类器导出支持
This commit is contained in:
parent
f3f2b38efb
commit
25ff7eb517
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue