Spaces:
Runtime error
Runtime error
JustinLin610
commited on
Commit
•
cd31959
1
Parent(s):
59fdcb0
debug
Browse files
app.py
CHANGED
@@ -70,7 +70,7 @@ def get_images(img: str, reader: ReaderLite, **kwargs):
|
|
70 |
return results
|
71 |
|
72 |
|
73 |
-
def draw_boxes(image, bounds, color='red', width=
|
74 |
draw = ImageDraw.Draw(image)
|
75 |
for i, bound in enumerate(bounds):
|
76 |
p0, p1, p2, p3 = bound
|
@@ -113,8 +113,8 @@ def patch_resize_transform(patch_image_size=480, is_document=False):
|
|
113 |
|
114 |
|
115 |
reader = ReaderLite(gpu=True)
|
116 |
-
overrides={"eval_cider": False, "beam":
|
117 |
-
"orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed":
|
118 |
models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
|
119 |
utils.split_paths('checkpoints/ocr_general_clean.pt'),
|
120 |
arg_overrides=overrides
|
@@ -165,7 +165,7 @@ def apply_half(t):
|
|
165 |
|
166 |
def ocr(img):
|
167 |
out_img = Image.open(img)
|
168 |
-
results = get_images(img, reader, link_threshold=0.2)
|
169 |
box_list, image_list = zip(*results)
|
170 |
draw_boxes(out_img, box_list)
|
171 |
|
|
|
70 |
return results
|
71 |
|
72 |
|
73 |
+
def draw_boxes(image, bounds, color='red', width=4):
|
74 |
draw = ImageDraw.Draw(image)
|
75 |
for i, bound in enumerate(bounds):
|
76 |
p0, p1, p2, p3 = bound
|
|
|
113 |
|
114 |
|
115 |
reader = ReaderLite(gpu=True)
|
116 |
+
overrides={"eval_cider": False, "beam": 5, "max_len_b": 64, "patch_image_size": 480,
|
117 |
+
"orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 42}
|
118 |
models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
|
119 |
utils.split_paths('checkpoints/ocr_general_clean.pt'),
|
120 |
arg_overrides=overrides
|
|
|
165 |
|
166 |
def ocr(img):
|
167 |
out_img = Image.open(img)
|
168 |
+
results = get_images(img, reader, text_confidence=0.8, text_threshold=0.5, link_threshold=0.2)
|
169 |
box_list, image_list = zip(*results)
|
170 |
draw_boxes(out_img, box_list)
|
171 |
|