feat: Added RegNet & ConvNext as supported models

app.py

@@ -17,7 +17,15 @@ from torchcam.utils import overlay_mask
 
 
 CAM_METHODS = ["CAM", "GradCAM", "GradCAMpp", "SmoothGradCAMpp", "ScoreCAM", "SSCAM", "ISCAM", "XGradCAM", "LayerCAM"]
-TV_MODELS = ["resnet18", "resnet50", "mobilenet_v2", "mobilenet_v3_small", "mobilenet_v3_large"]
+TV_MODELS = [
+    "resnet18",
+    "resnet50",
+    "mobilenet_v3_small",
+    "mobilenet_v3_large",
+    "regnet_y_400mf",
+    "convnext_tiny",
+    "convnext_small",
+]
 LABEL_MAP = requests.get(
     "https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
 ).json()

requirements.txt

@@ -1,3 +1,3 @@
 -e git+https://github.com/frgfm/torch-cam.git#egg=torchcam>0.3.0
 streamlit>=0.65.0
-torchvision>=0.8.0
+torchvision>=0.12.0,<1.0.0