add id

app.py

@@ -73,8 +73,9 @@ def get_images(img: str, reader: ReaderLite, **kwargs):
 
 def draw_boxes(image, bounds, color='red', width=2):
     draw = ImageDraw.Draw(image)
-    for bound in bounds:
+    for i, bound in enumerate(bounds):
         p0, p1, p2, p3 = bound
+        draw.text(p0, str(i+1), fill=color)
         draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
     return image
 
@@ -170,7 +171,7 @@ def ocr(img):
     draw_boxes(out_img, box_list)
 
     ocr_result = []
-    for box, image in zip(box_list, image_list):
+    for i, (box, image) in enumerate(zip(box_list, image_list)):
         image = Image.fromarray(image)
         sample = construct_sample(task, image, cfg.task.patch_image_size)
         sample = utils.move_to_cuda(sample) if use_cuda else sample
@@ -178,7 +179,7 @@ def ocr(img):
 
         with torch.no_grad():
             result, scores = eval_step(task, generator, models, sample)
-        ocr_result.append(result[0]['ocr'].replace(' ', ''))
+        ocr_result.append(str(i+1) + '\t' + result[0]['ocr'].replace(' ', ''))
 
     result = '\n'.join(ocr_result)