From 1b8f6ba0b6cdf39e23eea536fb3a9f4d6d76fcd5 Mon Sep 17 00:00:00 2001
From: OleehyO <1258009915@qq.com>
Date: Mon, 27 May 2024 23:19:35 +0800
Subject: [PATCH] bugfix: ocr_aug.py

Change "lhy_custom" in ink_swap_color to "random"
---
 src/models/ocr_model/utils/ocr_aug.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/models/ocr_model/utils/ocr_aug.py b/src/models/ocr_model/utils/ocr_aug.py
index c3e1d12..5678c61 100644
--- 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),