[feat] Add new pytorch db_resnet50 checkpoint

mindee · Jan 15, 2025 · cbe7cf8 · cbe7cf8
1 parent b0d2728
commit cbe7cf8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/using_doctr/using_models.rst b/docs/source/using_doctr/using_models.rst
@@ -57,7 +57,7 @@ For a comprehensive comparison, we have compiled a detailed benchmark on publicl
 +----------------+---------------------------------+-----------------+---------------+------------+---------------+------------+---------------+--------------------+
 | PyTorch        | db_resnet34                     | (1024, 1024, 3) | 22.4 M        | 82.76      | 76.75         | 89.20      | 71.74         | 0.8                |
 +----------------+---------------------------------+-----------------+---------------+------------+---------------+------------+---------------+--------------------+
-| PyTorch        | db_resnet50                     | (1024, 1024, 3) | 25.4 M        | 83.56      | 86.68         | 92.61      | 86.39         | 1.1                |
+| PyTorch        | db_resnet50                     | (1024, 1024, 3) | 25.4 M        | 84.55      | 89.64         | 93.73      | 86.27         | 1.1                |
 +----------------+---------------------------------+-----------------+---------------+------------+---------------+------------+---------------+--------------------+
 | PyTorch        | db_mobilenet_v3_large           | (1024, 1024, 3) | 4.2 M         | 82.69      | 84.63         | 94.51      | 70.28         | 0.5                |
 +----------------+---------------------------------+-----------------+---------------+------------+---------------+------------+---------------+--------------------+

diff --git a/doctr/models/detection/differentiable_binarization/pytorch.py b/doctr/models/detection/differentiable_binarization/pytorch.py
@@ -28,7 +28,7 @@
         "input_shape": (3, 1024, 1024),
         "mean": (0.798, 0.785, 0.772),
         "std": (0.264, 0.2749, 0.287),
-        "url": "https://doctr-static.mindee.com/models?id=v0.7.0/db_resnet50-79bd7d70.pt&src=0",
+        "url": "https://github.com/mindee/doctr/releases/download/v0.10.0/db_resnet50-a5b57aac.pt",
     },
     "db_resnet34": {
         "input_shape": (3, 1024, 1024),