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 1nhT7yuqJ8V8vfoJdI0Qn6notmoLFRrmm')
 os.system('pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"')
 os.system('python setup.py build develop --user')
+os.system('pip install --upgrade --no-cache-dir gdown')
+os.system('gdown -O ./output/ctw/model_ctw.pth 1nhT7yuqJ8V8vfoJdI0Qn6notmoLFRrmm')
 #os.system('pip install git+https://github.com/Cyril-Sterling/ContourNet')
 
 import cv2
@@ -24,8 +24,8 @@ def infer(filepath):
         output_polygon=True
     )
     image = cv2.imread(filepath)
-    result_polygons, result_masks = det_demo.run_on_opencv_image(image)
-    image = det_demo.visualization(image, result_polygons, result_masks)
+    result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
+    image = det_demo.visualization(image, result_polygons, result_masks, result_boxes)
     cv2.imwrite('result.jpg', image)
     return 'result.jpg'#, pd.DataFrame(result_words)
 

det_demo.py

@@ -499,7 +499,7 @@ class DetDemo(object):
         cv2.imwrite('./demo/example_results.jpg', im_write[:, :, ::-1])
         '''
         
-        return contours, np.array(masks.repeat(1,3,1,1)).astype(np.bool_).transpose(0,2,3,1)
+        return contours, np.array(masks.repeat(1,3,1,1)).astype(np.bool_).transpose(0,2,3,1), boxes
 
     def process_char_mask(self, char_masks, boxes, threshold=192):
         texts, rec_scores = [], []
@@ -580,6 +580,8 @@ class DetDemo(object):
         green[...,2] = 0
         for mask in masks:
             image[mask] = image[mask] * 0.5 + green[mask] * 0.5
+        for box in boxes:
+            cv2.rectangle(image,box,True,(0,0,255))
         '''
         for polygon in polygons:
             pts = np.array(polygon, np.int32)