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

rename

上级 8f095942
...@@ -65,7 +65,7 @@ class BaseClassifier(BaseAPI): ...@@ -65,7 +65,7 @@ class BaseClassifier(BaseAPI):
inputs = OrderedDict([('image', image)]) inputs = OrderedDict([('image', image)])
outputs = OrderedDict([('predict', softmax_out)]) outputs = OrderedDict([('predict', softmax_out)])
if mode == 'test': if mode == 'test':
self.explanation_feats = OrderedDict([('logits', net_out)]) self.interpretation_feats = OrderedDict([('logits', net_out)])
if mode != 'test': if mode != 'test':
cost = fluid.layers.cross_entropy(input=softmax_out, label=label) cost = fluid.layers.cross_entropy(input=softmax_out, label=label)
avg_cost = fluid.layers.mean(cost) avg_cost = fluid.layers.mean(cost)
......
...@@ -25,5 +25,5 @@ def interpretation_predict(model, images): ...@@ -25,5 +25,5 @@ def interpretation_predict(model, images):
result = model.exe.run( result = model.exe.run(
model.test_prog, model.test_prog,
feed={'image': new_imgs}, feed={'image': new_imgs},
fetch_list=list(model.explanation_feats.values())) fetch_list=list(model.interpretation_feats.values()))
return result 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.
先完成此消息的编辑!
想要评论请 注册