First model version

app.py

@@ -1,8 +1,8 @@
 import os
 os.system('pip install --upgrade --no-cache-dir gdown')
 os.system('gdown -O ./output/ctw/model_ctw.pth 1Ajslu_9WisuZ2nJGzE6qbD87aK6_ozzA')
-os.system('gdown -O ./workdir.zip 1mYM_26qHUom_5NU7iutHneB_KHlLjL5y')
-os.system('unzip workdir.zip')
+#os.system('gdown -O ./workdir.zip 1mYM_26qHUom_5NU7iutHneB_KHlLjL5y')
+#os.system('unzip workdir.zip')
 os.system('pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"')
 os.system('python setup.py build develop --user')
 
@@ -17,12 +17,13 @@ from maskrcnn_benchmark.config import cfg
 
 def process_image(filepath):
   # rec model
+  '''
   config = Config('configs/rec/train_abinet.yaml')
   config.model_vision_checkpoint = None
   model = get_model(config)
   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 model
   cfg.merge_from_file('./configs/det/r50_baseline.yaml')
   cfg.merge_from_list(["MODEL.DEVICE", "cpu"])
@@ -38,7 +39,11 @@ def process_image(filepath):
   result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
   
   # cut patch
-  image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+  #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 = [preprocess(patch, config.dataset_image_width, config.dataset_image_height) for patch in patchs]
   patchs = torch.stack(patchs, dim=0)
@@ -46,6 +51,7 @@ def process_image(filepath):
   res = model(patchs)
   rec_result = postprocess(res, charset, 'alignment')[0]
   print(rec_result)
+  '''
   
   # visual detect results
   visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
@@ -64,7 +70,7 @@ description = "西北工业大学航海学院张博强毕设，目前识别部
 
 iface = gr.Interface(fn=process_image, 
                      inputs=[gr.inputs.Image(label="image", type="filepath")], 
-                     outputs=[gr.outputs.Image(), gr.outputs.Textbox()],
+                     outputs=[gr.outputs.Image()],#, gr.outputs.Textbox()
                      title=title,
                      description=description,
                      examples=glob.glob('figs/test/*.png'))