diff --git a/.gitignore b/.gitignore index 787a849..789f903 100644 --- a/.gitignore +++ b/.gitignore @@ -5,4 +5,5 @@ **/logs **/.cache **/tmp* -**/data \ No newline at end of file +**/data +**/ckpt \ No newline at end of file diff --git a/src/models/ocr_model/utils/transforms.py b/src/models/ocr_model/utils/transforms.py index bdb5ef9..5bf21a0 100644 --- a/src/models/ocr_model/utils/transforms.py +++ b/src/models/ocr_model/utils/transforms.py @@ -16,7 +16,7 @@ from ...globals import ( MAX_RESIZE_RATIO, MIN_RESIZE_RATIO ) -train_pipeline = default_augraphy_pipeline() +train_pipeline = default_augraphy_pipeline(scan_only=True) general_transform_pipeline = v2.Compose([ v2.ToImage(), # Convert to tensor, only needed if you had a PIL image