From aaee57acd264b47322f1be854f0fbe1a81312b2c 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: Tue, 9 Apr 2024 13:50:35 +0000
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BA=86=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=A2=9E=E5=BC=BA=E7=9A=84=E6=A6=82=E7=8E=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/models/ocr_model/utils/ocr_aug.py | 32 ++++++++++++++++++---------
 src/start_web.sh                      |  3 ++-
 2 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/src/models/ocr_model/utils/ocr_aug.py b/src/models/ocr_model/utils/ocr_aug.py
index 0e364ca..da48a70 100644
--- a/src/models/ocr_model/utils/ocr_aug.py
+++ b/src/models/ocr_model/utils/ocr_aug.py
@@ -16,7 +16,8 @@ def ocr_augmentation_pipeline():
             ink_swap_max_height_range=(100, 120),
             ink_swap_min_area_range=(10, 20),
             ink_swap_max_area_range=(400, 500),
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         LinesDegradation(
             line_roi=(0.0, 0.0, 1.0, 1.0),
@@ -28,7 +29,8 @@ def ocr_augmentation_pipeline():
             line_long_to_short_ratio=(5, 7),
             line_replacement_probability=(0.4, 0.5),
             line_replacement_thickness=(1, 3),
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
 
         #  ============================
@@ -44,7 +46,8 @@ def ocr_augmentation_pipeline():
                     severity=(0.4, 0.6),
                 ),
             ],
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         #  ============================
 
@@ -56,7 +59,8 @@ def ocr_augmentation_pipeline():
             blur_kernel_size=(5, 5),
             blur_sigma=0,
             noise_type="perlin",
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         #  ============================
 
@@ -68,12 +72,14 @@ def ocr_augmentation_pipeline():
             turbulence_range=(2, 5),
             texture_width_range=(300, 500),
             texture_height_range=(300, 500),
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         BrightnessTexturize(  # tested
             texturize_range=(0.9, 0.99),
             deviation=0.03,
-            p=0.2
+            # p=0.2
+            p=0.4
         )
     ]
 
@@ -84,7 +90,8 @@ def ocr_augmentation_pipeline():
             color_shift_iterations=(2, 3),
             color_shift_brightness_range=(0.9, 1.1),
             color_shift_gaussian_kernel_range=(3, 3),
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
 
         DirtyDrum(  # tested
@@ -95,7 +102,8 @@ def ocr_augmentation_pipeline():
             noise_value=(64, 224),
             ksize=random.choice([(3, 3), (5, 5), (7, 7)]),
             sigmaX=0,
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
 
         # =====================================
@@ -119,7 +127,8 @@ def ocr_augmentation_pipeline():
                     gamma_range=(0.9, 1.1),
                 ),
             ],
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         # =====================================
 
@@ -130,10 +139,11 @@ def ocr_augmentation_pipeline():
                     subtle_range=random.randint(5, 10),
                 ),
                 Jpeg(
-                    quality_range=(85, 95),
+                    quality_range=(70, 95),
                 ),
             ],
-            p=0.2
+            # p=0.2
+            p=0.4
         ),
         # =====================================
     ]
diff --git a/src/start_web.sh b/src/start_web.sh
index 7475147..3d700cc 100755
--- a/src/start_web.sh
+++ b/src/start_web.sh
@@ -1,7 +1,8 @@
 #!/usr/bin/env bash
 set -exu
 
-export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/model/ckpt"
+# export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/model/ckpt"
+export CHECKPOINT_DIR="default"
 export TOKENIZER_DIR="/home/lhy/code/TexTeller/src/models/tokenizer/roberta-tokenizer-7Mformulas"
 export USE_CUDA=True  # True or False (case-sensitive)
 export NUM_BEAM=3