sunyanfang01 5 years ago
parent
commit
15edefbb21
2 changed files with 2 additions and 2 deletions
  1. 1 1
      paddlex/cv/models/classifier.py
  2. 1 1
      paddlex/interpret/interpretation_predict.py

+ 1 - 1
paddlex/cv/models/classifier.py

@@ -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)

+ 1 - 1
paddlex/interpret/interpretation_predict.py

@@ -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