提交 15edefbb 编写于 作者: S sunyanfang01

rename

上级 8f095942
......@@ -65,7 +65,7 @@ class BaseClassifier(BaseAPI):
inputs = OrderedDict([('image', image)])
outputs = OrderedDict([('predict', softmax_out)])
if mode == 'test':
self.explanation_feats = OrderedDict([('logits', net_out)])
self.interpretation_feats = OrderedDict([('logits', net_out)])
if mode != 'test':
cost = fluid.layers.cross_entropy(input=softmax_out, label=label)
avg_cost = fluid.layers.mean(cost)
......
......@@ -25,5 +25,5 @@ def interpretation_predict(model, images):
result = model.exe.run(
model.test_prog,
feed={'image': new_imgs},
fetch_list=list(model.explanation_feats.values()))
fetch_list=list(model.interpretation_feats.values()))
return result
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册