nelbarman053
commited on
Commit
•
2cdf2cf
1
Parent(s):
18b304b
Figure size changed
Browse files- app.py +4 -4
- xai/xai_visualization.png +0 -0
app.py
CHANGED
@@ -17,9 +17,9 @@ from torchvision import transforms
|
|
17 |
from pytorch_grad_cam import GradCAM
|
18 |
from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
|
19 |
|
20 |
-
import pathlib
|
21 |
-
temp = pathlib.PosixPath
|
22 |
-
pathlib.PosixPath = pathlib.WindowsPath
|
23 |
|
24 |
model = load_learner("models/recognizer_model.pkl")
|
25 |
|
@@ -38,7 +38,7 @@ def xai_visualization(image, image_tensor, targeted_category, model, target_laye
|
|
38 |
|
39 |
mask = grayscale_cam[0, :]
|
40 |
|
41 |
-
plt.figure(figsize=(
|
42 |
|
43 |
plt.axis('off')
|
44 |
|
|
|
17 |
from pytorch_grad_cam import GradCAM
|
18 |
from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
|
19 |
|
20 |
+
# import pathlib
|
21 |
+
# temp = pathlib.PosixPath
|
22 |
+
# pathlib.PosixPath = pathlib.WindowsPath
|
23 |
|
24 |
model = load_learner("models/recognizer_model.pkl")
|
25 |
|
|
|
38 |
|
39 |
mask = grayscale_cam[0, :]
|
40 |
|
41 |
+
plt.figure(figsize=(10,5))
|
42 |
|
43 |
plt.axis('off')
|
44 |
|
xai/xai_visualization.png
CHANGED