First model version

app.py

@@ -36,11 +36,12 @@ def infer(filepath):
     )
     
     image = cv2.imread(filepath)
+    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]
     patchs = [preprocess(patch, config.dataset_image_width, config.dataset_image_height) for patch in patchs]
     patchs = torch.stack(patchs, dim=0)
     
-    result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
     visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
     
     cv2.imwrite('result.jpg', visual_image)