khan994 commited on
Commit
c512393
·
1 Parent(s): 4a952f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def gradcam(img_create):
47
  cam_map = torch.einsum('ck,kij->cij', learn.model[1][-1].weight, new_act)
48
 
49
  x_dec = TensorImage(dls_augmented.train.decode((x,))[0][0])
50
- plt.rcParams['figure.figsize'] = (128, 128)
51
  fig,ax = plt.subplots()
52
  x_dec.show(ctx=ax)
53
  ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,128,128,0), interpolation='bilinear', cmap='magma');
 
47
  cam_map = torch.einsum('ck,kij->cij', learn.model[1][-1].weight, new_act)
48
 
49
  x_dec = TensorImage(dls_augmented.train.decode((x,))[0][0])
50
+ #plt.rcParams['figure.figsize'] = (128, 128)
51
  fig,ax = plt.subplots()
52
  x_dec.show(ctx=ax)
53
  ax.imshow(cam_map[1].detach().cpu(), alpha=0.6, extent=(0,128,128,0), interpolation='bilinear', cmap='magma');