From b293026f5e5f3d0dafbede590a0fa2058a7a27c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E6=B4=8B=E4=B8=89=E6=B4=8B?= <1258009915@qq.com>
Date: Wed, 13 Mar 2024 02:55:28 +0000
Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=A2=9E=E5=BC=BA=E5=8A=A0?=
 =?UTF-8?q?=E5=85=A5=E4=BA=86=E6=98=AF=E5=90=A6=E5=8F=AA=E9=92=88=E5=AF=B9?=
 =?UTF-8?q?=E6=89=AB=E6=8F=8F=E4=BB=B6=E7=9A=84=E9=80=89=E9=A1=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore                               | 3 ++-
 src/models/ocr_model/utils/transforms.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 787a849..789f903 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,4 +5,5 @@
 **/logs
 **/.cache
 **/tmp*
-**/data
\ No newline at end of file
+**/data
+**/ckpt
\ No newline at end of file
diff --git a/src/models/ocr_model/utils/transforms.py b/src/models/ocr_model/utils/transforms.py
index bdb5ef9..5bf21a0 100644
--- a/src/models/ocr_model/utils/transforms.py
+++ b/src/models/ocr_model/utils/transforms.py
@@ -16,7 +16,7 @@ from ...globals import (
     MAX_RESIZE_RATIO, MIN_RESIZE_RATIO
 )
 
-train_pipeline = default_augraphy_pipeline()
+train_pipeline = default_augraphy_pipeline(scan_only=True)
 
 general_transform_pipeline = v2.Compose([
     v2.ToImage(),    # Convert to tensor, only needed if you had a PIL image