feat: ✨ torch mobilenet_v3_small_orientation chkpt

mindee · Apr 22, 2024 · 68af248 · 68af248
1 parent 0285711
commit 68af248
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py
@@ -56,7 +56,7 @@
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 128, 128),
         "classes": [0, -90, 180, 90],
-        "url": "https://doctr-static.mindee.com/models?id=v0.4.1/classif_mobilenet_v3_small-24f8ff57.pt&src=0",
+        "url": "https://doctr-static.mindee.com/models?id=v0.8.1/mobilenet_v3_small_orientation-97a3028f.pt&src=0",
     },
     "mobilenet_v3_small_page_orientation": {
         "mean": (0.694, 0.695, 0.693),