diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py index 73b98d833f..2f040431b0 100644 --- 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),