First model version

app.py

@@ -22,7 +22,7 @@ import torch
 def infer(filepath):
     config = Config('configs/rec/train_abinet.yaml')
     config.model_vision_checkpoint = None
-    model = get_model(config)
+    model = get_model(config).to_device('cuda')
     model = load(model, 'workdir/train-abinet/best-train-abinet.pth')
     charset = CharsetMapper(filename=config.dataset_charset_path, max_length=config.dataset_max_length + 1)
     

det_demo.py

@@ -580,7 +580,7 @@ class DetDemo(object):
         green[...,2] = 0
         for mask, word, box in zip(masks, words, boxes):
             image[mask] = image[mask] * 0.5 + green[mask] * 0.5
-            cv2.putText(image, word, (box[0], box[1]), cv2.FONT_HERSHEY_COMPLEX, 0.4, (0,0,255), 1)
+            cv2.putText(image, word, (box[0], box[1]), cv2.FONT_HERSHEY_COMPLEX, 0.6, (0,0,255), 1)
         '''
         for box in boxes:
             cv2.rectangle(image,(box[0], box[1]), (box[2], box[3]), (0,0,255), 2)