Update app.py
Browse files
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');
|