First model version

app.py

@@ -8,8 +8,8 @@ os.system('python setup.py build develop --user')
 
 import glob
 import gradio as gr
-from demo import get_model, preprocess, postprocess, load
-from utils import Config, Logger, CharsetMapper
+#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
@@ -40,9 +40,9 @@ def process_image(filepath):
   
   # 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 = [cv2.resize(patch, (128,32)) for patch in patchs]
-  patchs = np.stack(patchs, axis=0).transpose(0,3,1,2)
+  #patchs = [image[box[1]:box[3], box[0]:box[2], :] for box in result_boxes]
+  #patchs = [cv2.resize(patch, (128,32)) for patch in patchs]
+  #patchs = np.stack(patchs, axis=0).transpose(0,3,1,2)
   '''
   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]
@@ -57,12 +57,6 @@ def process_image(filepath):
   visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
   cv2.imwrite('result.jpg', visual_image)
   return 'result.jpg'#, pd.DataFrame(result_words)
-  
-
-  img = image.convert('RGB')
-  img = preprocess(img, config.dataset_image_width, config.dataset_image_height)
-  res = model(img)
-  return postprocess(res, charset, 'alignment')[0][0]
 
 title = "张博强毕设中期展示（文本识别部分）"
 description = "西北工业大学航海学院张博强毕设，目前识别部分进度为复现abinet，本网页为abinet复现的可视化web端展示"