diff --git a/doctr/models/detection/fast/base.py b/doctr/models/detection/fast/base.py index dcea87549..10ce7d302 100644 --- a/doctr/models/detection/fast/base.py +++ b/doctr/models/detection/fast/base.py @@ -153,7 +153,7 @@ class _FAST(BaseModel): min_size_box: int = 3 assume_straight_pages: bool = True - shrink_ratio = 0.1 + shrink_ratio = 0.4 def build_target( self, diff --git a/doctr/models/detection/fast/pytorch.py b/doctr/models/detection/fast/pytorch.py index e5b7bcfc9..3e3ed3f0e 100644 --- a/doctr/models/detection/fast/pytorch.py +++ b/doctr/models/detection/fast/pytorch.py @@ -122,7 +122,7 @@ def __init__( bin_thresh: float = 0.3, box_thresh: float = 0.1, dropout_prob: float = 0.1, - pooling_size: int = 9, + pooling_size: int = 4, assume_straight_pages: bool = True, exportable: bool = False, cfg: Optional[Dict[str, Any]] = {},