F-G Fernandez
commited on
Commit
·
a9a3664
1
Parent(s):
4e260f9
feat: Added RegNet & ConvNext as supported models
Browse files- app.py +9 -1
- requirements.txt +1 -1
app.py
CHANGED
@@ -17,7 +17,15 @@ from torchcam.utils import overlay_mask
|
|
17 |
|
18 |
|
19 |
CAM_METHODS = ["CAM", "GradCAM", "GradCAMpp", "SmoothGradCAMpp", "ScoreCAM", "SSCAM", "ISCAM", "XGradCAM", "LayerCAM"]
|
20 |
-
TV_MODELS = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
LABEL_MAP = requests.get(
|
22 |
"https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
|
23 |
).json()
|
|
|
17 |
|
18 |
|
19 |
CAM_METHODS = ["CAM", "GradCAM", "GradCAMpp", "SmoothGradCAMpp", "ScoreCAM", "SSCAM", "ISCAM", "XGradCAM", "LayerCAM"]
|
20 |
+
TV_MODELS = [
|
21 |
+
"resnet18",
|
22 |
+
"resnet50",
|
23 |
+
"mobilenet_v3_small",
|
24 |
+
"mobilenet_v3_large",
|
25 |
+
"regnet_y_400mf",
|
26 |
+
"convnext_tiny",
|
27 |
+
"convnext_small",
|
28 |
+
]
|
29 |
LABEL_MAP = requests.get(
|
30 |
"https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
|
31 |
).json()
|
requirements.txt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
-e git+https://github.com/frgfm/torch-cam.git#egg=torchcam>0.3.0
|
2 |
streamlit>=0.65.0
|
3 |
-
torchvision>=0.
|
|
|
1 |
-e git+https://github.com/frgfm/torch-cam.git#egg=torchcam>0.3.0
|
2 |
streamlit>=0.65.0
|
3 |
+
torchvision>=0.12.0,<1.0.0
|