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

数据增强加入了是否只针对扫描件的选项

Browse Source
This commit is contained in:
三洋三洋
2024-03-13 02:55:28 +00:00
parent 2d6c46b88d
commit b293026f5e
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -6,3 +6,4 @@
**/.cache **/.cache
**/tmp* **/tmp*
**/data **/data
**/ckpt

2
src/models/ocr_model/utils/transforms.py
View File

@@ -16,7 +16,7 @@ from ...globals import (
MAX_RESIZE_RATIO, MIN_RESIZE_RATIO MAX_RESIZE_RATIO, MIN_RESIZE_RATIO
) )
train_pipeline = default_augraphy_pipeline() train_pipeline = default_augraphy_pipeline(scan_only=True)
general_transform_pipeline = v2.Compose([ general_transform_pipeline = v2.Compose([
v2.ToImage(), # Convert to tensor, only needed if you had a PIL image v2.ToImage(), # Convert to tensor, only needed if you had a PIL image