debug

app.py

@@ -70,7 +70,7 @@ def get_images(img: str, reader: ReaderLite, **kwargs):
     return results
 
 
-def draw_boxes(image, bounds, color='red', width=10):
+def draw_boxes(image, bounds, color='red', width=4):
     draw = ImageDraw.Draw(image)
     for i, bound in enumerate(bounds):
         p0, p1, p2, p3 = bound
@@ -113,8 +113,8 @@ def patch_resize_transform(patch_image_size=480, is_document=False):
 
 
 reader = ReaderLite(gpu=True)
-overrides={"eval_cider": False, "beam": 4, "max_len_b": 32, "patch_image_size": 480,
-           "orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 7}
+overrides={"eval_cider": False, "beam": 5, "max_len_b": 64, "patch_image_size": 480,
+           "orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 42}
 models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
     utils.split_paths('checkpoints/ocr_general_clean.pt'),
     arg_overrides=overrides
@@ -165,7 +165,7 @@ def apply_half(t):
 
 def ocr(img):
     out_img = Image.open(img)
-    results = get_images(img, reader, link_threshold=0.2)
+    results = get_images(img, reader, text_confidence=0.8, text_threshold=0.5, link_threshold=0.2)
     box_list, image_list = zip(*results)
     draw_boxes(out_img, box_list)