andy-wyx commited on
Commit
7187e08
·
1 Parent(s): 5579c05

debugging: xai output distortion

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -211,7 +211,8 @@ def explain_image(input_image,model_name,explain_method,nb_samples):
211
  else:
212
  size = 600
213
  #saliency, integrated, smoothgrad,
214
- classes,exp_list = explain(model,input_image,explain_method,nb_samples,size = size, n_classes=n_classes)
 
215
  #original = saliency + integrated + smoothgrad
216
  print('done')
217
 
 
211
  else:
212
  size = 600
213
  #saliency, integrated, smoothgrad,
214
+ h,w = input_image.shape[:2]
215
+ classes,exp_list = explain(model,input_image,h,w,explain_method,nb_samples,size = size, n_classes=n_classes)
216
  #original = saliency + integrated + smoothgrad
217
  print('done')
218