andy-wyx commited on
Commit
9c38311
·
1 Parent(s): f8b140a

update image padding for xai

Browse files
Files changed (1) hide show
  1. explanations.py +1 -1
explanations.py CHANGED
@@ -110,7 +110,7 @@ def explain(model, input_image,explain_method,nb_samples,size=600, n_classes=171
110
  if len(phi.shape) == 3:
111
  phi = np.mean(phi, -1)
112
  show(X[0],output_size = size)
113
- show(phi, p=1, alpha=0.4, output_size = size)
114
  # show(X[0][:,size_repetitions:2*size_repetitions,:])
115
  # show(phi[:,size_repetitions:2*size_repetitions], p=1, alpha=0.4)
116
  plt.savefig(f'phi_{e}{i}.png')
 
110
  if len(phi.shape) == 3:
111
  phi = np.mean(phi, -1)
112
  show(X[0],output_size = size)
113
+ show(phi, output_size = size,p=1, alpha=0.4, )
114
  # show(X[0][:,size_repetitions:2*size_repetitions,:])
115
  # show(phi[:,size_repetitions:2*size_repetitions], p=1, alpha=0.4)
116
  plt.savefig(f'phi_{e}{i}.png')