rise

explanations.py

@@ -66,17 +66,20 @@ def explain(model, input_image,size=600, n_classes=171) :
     explanations = []
     for e,explainer in enumerate(explainers):
         print(f'{e}/{len(explainers)}')
+        print('1')
         Y = tf.one_hot([Y], n_classes)
         phi = np.abs(explainer(X, Y))[0]
-
+        print('1')
         if len(phi.shape) == 3:
             phi = np.mean(phi, -1)
-        
+        print('1')
         show(X[0][:,size_repetitions:2*size_repetitions,:])
         show(phi[:,size_repetitions:2*size_repetitions], p=1, alpha=0.4)
-        
+        print('1')
         plt.savefig(f'phi_{e}.png')
+        print('1')
         explanations.append(f'phi_{e}.png')
+        print('1')
         print(type(explanations))
         print(len(explanations))