fix: target_ar split crops (#681)

mindee · Dec 7, 2021 · d9f432d · d9f432d
1 parent e076418
commit d9f432d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/docs/source/using_models.rst b/docs/source/using_models.rst
@@ -94,8 +94,8 @@ For a comprehensive comparison, we have compiled a detailed benchmark on publicl
    * - crnn_vgg16_bn
      - (32, 128, 3)
      - 15.8M
-     - 87.15
-     - 92.92
+     - 87.18
+     - 92.93
      - 12.8
    * - crnn_mobilenet_v3_small
      - (32, 128, 3)
@@ -165,7 +165,7 @@ For a comprehensive comparison, we have compiled a detailed benchmark on publicl
 +========================================+============+===============+=========+============+===============+=========+
 | **Architecture**                       | **Recall** | **Precision** | **FPS** | **Recall** | **Precision** | **FPS** |
 +----------------------------------------+------------+---------------+---------+------------+---------------+---------+
-| db_resnet50 + crnn_vgg16_bn            | 71.00      | 76.02         | 0.85    | 83.87      |   81.34       | 1.6     |
+| db_resnet50 + crnn_vgg16_bn            | 71.25      | 76.02         | 0.85    | 84.00      |   81.42       | 1.6     |
 +----------------------------------------+------------+---------------+---------+------------+---------------+---------+
 | db_resnet50 + master                   | 71.03      | 76.06         |         | 84.49      |   81.94       |         |
 +----------------------------------------+------------+---------------+---------+------------+---------------+---------+

diff --git a/doctr/models/recognition/predictor/pytorch.py b/doctr/models/recognition/predictor/pytorch.py
@@ -38,7 +38,7 @@ def __init__(
         self.split_wide_crops = split_wide_crops
         self.critical_ar = 8  # Critical aspect ratio
         self.dil_factor = 1.4  # Dilation factor to overlap the crops
-        self.target_ar = 4  # Target aspect ratio
+        self.target_ar = 6  # Target aspect ratio
 
     @torch.no_grad()
     def forward(

diff --git a/doctr/models/recognition/predictor/tensorflow.py b/doctr/models/recognition/predictor/tensorflow.py
@@ -41,7 +41,7 @@ def __init__(
         self.split_wide_crops = split_wide_crops
         self.critical_ar = 8  # Critical aspect ratio
         self.dil_factor = 1.4  # Dilation factor to overlap the crops
-        self.target_ar = 4  # Target aspect ratio
+        self.target_ar = 6  # Target aspect ratio
 
     def __call__(
         self,