Cyril666 commited on
Commit
5f9ac67
·
1 Parent(s): bdb0dcc

First model version

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -42,6 +42,9 @@ def infer(filepath):
42
  patchs = [image[box[1]:box[3], box[0]:box[2], :] for box in result_boxes]
43
  patchs = [preprocess(patch, config.dataset_image_width, config.dataset_image_height) for patch in patchs]
44
  patchs = torch.stack(patchs, dim=0)
 
 
 
45
 
46
  visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
47
 
 
42
  patchs = [image[box[1]:box[3], box[0]:box[2], :] for box in result_boxes]
43
  patchs = [preprocess(patch, config.dataset_image_width, config.dataset_image_height) for patch in patchs]
44
  patchs = torch.stack(patchs, dim=0)
45
+ res = model(patchs)
46
+ rec_result = postprocess(res, charset, 'alignment')[0]
47
+ print(rec_result)
48
 
49
  visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
50