MedCat
/

table_detection_3

Model card Files Files and versions Community

duytrq commited on May 4

Commit

f34343b

•

1 Parent(s): 7eb8ca1

Update config.py

Files changed (1) hide show

config.py +1 -1

config.py CHANGED Viewed

@@ -18,7 +18,7 @@ class Config(f.Pipe):
             YOLO_IMAGE_SIZE = 640,
             HEIGHT_EXPAND_RATIO = 0.03,
-            USE_STAMP_DETECTION = False,
             STAMP_DETECTION_CHECKPOINT = 'MedCat/checkpoints:stamp_detection_yolo_v1.pt',
             ORIGINAL_STAMP_WEIGHT_PATH = 'https://drive.google.com/file/d/1YHH7pLoZEdyxw2AoLz9G4lrq6uuxweYB/view?usp=sharing',
             STAMP_REMOVING_IOU_THRESHOLD = 0.6

             YOLO_IMAGE_SIZE = 640,
             HEIGHT_EXPAND_RATIO = 0.03,
+            USE_STAMP_DETECTION = True,
             STAMP_DETECTION_CHECKPOINT = 'MedCat/checkpoints:stamp_detection_yolo_v1.pt',
             ORIGINAL_STAMP_WEIGHT_PATH = 'https://drive.google.com/file/d/1YHH7pLoZEdyxw2AoLz9G4lrq6uuxweYB/view?usp=sharing',
             STAMP_REMOVING_IOU_THRESHOLD = 0.6