feat: ✨ torch mobilenet_v3_small_orientation chkpt (mindee#1557)

odulcy-mindee · Apr 22, 2024 · 248afe8 · 248afe8
1 parent 68e7388
commit 248afe8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py
@@ -54,9 +54,9 @@
     "mobilenet_v3_small_crop_orientation": {
         "mean": (0.694, 0.695, 0.693),
         "std": (0.299, 0.296, 0.301),
-        "input_shape": (3, 128, 128),
+        "input_shape": (3, 256, 256),
         "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),