First model version

app.py

@@ -40,6 +40,7 @@ def infer(img):
     )
     
     image = cv2.imread(filepath)
+    print(image.shape)
     result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
     
     patchs = [image[box[1]:box[3], box[0]:box[2], :] for box in result_boxes]
@@ -50,6 +51,7 @@ def infer(img):
     
     visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes, result_words)
     
+    print(visual_image.shape)
     cv2.imwrite('result.jpg', visual_image)
     return ['result.jpg', pd.DataFrame(result_words)]