Update app.py

app.py

@@ -24,7 +24,7 @@ model_path = 'model_final.pth'
 
 cfg = get_cfg()
 cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.48
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
 cfg.MODEL.WEIGHTS = model_path
 
@@ -40,7 +40,8 @@ def inference(image):
 
     height = image.height
 
-    img = np.array(image.resize((500, height)))
+    # img = np.array(image.resize((500, height)))
+    img = np.array(image)
     outputs = predictor(img)
     v = Visualizer(img[:, :, ::-1],
                    metadata=my_metadata,