bugfix: ocr_aug.py

Change "lhy_custom" in ink_swap_color to "random"
2024-05-27 23:19:35 +08:00
parent 0a51bde1c5
commit 19827f1837
1 changed files with 1 additions and 2 deletions
--- a/src/models/ocr_model/utils/ocr_aug.py
+++ b/src/models/ocr_model/utils/ocr_aug.py
@@ -3,12 +3,11 @@ import random

 def ocr_augmentation_pipeline():
    pre_phase = [
-        # Rescale(scale="optimal", target_dpi = 300,  p = 1.0),
    ]

    ink_phase = [
        InkColorSwap(
-            ink_swap_color="lhy_custom",
+            ink_swap_color="random",
            ink_swap_sequence_number_range=(5, 10),
            ink_swap_min_width_range=(2, 3),
            ink_swap_max_width_range=(100, 120),