From 68af2487e0417aba8e72ba55fd0958156e626028 Mon Sep 17 00:00:00 2001 From: Olivier Dulcy Date: Wed, 17 Apr 2024 17:20:08 +0200 Subject: [PATCH 1/2] feat: :sparkles: torch `mobilenet_v3_small_orientation` chkpt --- doctr/models/classification/mobilenet/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py index 73b98d833f..90707ce3ef 100644 --- 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), From f142218478b8419b958a4f41e9bc0e06247ae0bf Mon Sep 17 00:00:00 2001 From: Olivier Dulcy Date: Mon, 22 Apr 2024 17:00:12 +0200 Subject: [PATCH 2/2] input shape --- doctr/models/classification/mobilenet/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py index 90707ce3ef..2f040431b0 100644 --- a/doctr/models/classification/mobilenet/pytorch.py +++ b/doctr/models/classification/mobilenet/pytorch.py @@ -54,7 +54,7 @@ "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.8.1/mobilenet_v3_small_orientation-97a3028f.pt&src=0", },