First model version

app.py

@@ -10,8 +10,12 @@ import glob
 import gradio as gr
 from demo import get_model, preprocess, postprocess, load
 from utils import Config, Logger, CharsetMapper
+import cv2
+import pandas as pd
+from det_demo import DetDemo
+from maskrcnn_benchmark.config import cfg
 
-def process_image(image):
+def process_image(filepath):
   # rec model
   config = Config('configs/rec/train_abinet.yaml')
   config.model_vision_checkpoint = None
@@ -34,6 +38,7 @@ def process_image(image):
   result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
   
   # cut patch
+  image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
   patchs = [image[box[1]:box[3], box[0]:box[2], :] for box in result_boxes]
   patchs = [preprocess(patch, config.dataset_image_width, config.dataset_image_height) for patch in patchs]
   patchs = torch.stack(patchs, dim=0)