YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix: ocr_aug.py

Browse Source 
Change "lhy_custom" in ink_swap_color to "random"
This commit is contained in:
OleehyO
2024-05-27 23:19:35 +08:00
committed by 三洋三洋
parent 0a51bde1c5
commit 19827f1837
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/models/ocr_model/utils/ocr_aug.py
View File

@@ -3,12 +3,11 @@ import random
def ocr_augmentation_pipeline(): def ocr_augmentation_pipeline():
pre_phase = [ pre_phase = [
# Rescale(scale="optimal", target_dpi = 300, p = 1.0),
] ]
ink_phase = [ ink_phase = [
InkColorSwap( InkColorSwap(
ink_swap_color="lhy_custom", ink_swap_color="random",
ink_swap_sequence_number_range=(5, 10), ink_swap_sequence_number_range=(5, 10),
ink_swap_min_width_range=(2, 3), ink_swap_min_width_range=(2, 3),
ink_swap_max_width_range=(100, 120), ink_swap_max_width_range=(100, 120),