update app

app.py

@@ -20,7 +20,7 @@ from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
 
 
-model_name='./configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x_4gpu_transfiner.yaml'
+model_name='./configs/transfiner/mask_rcnn_R_50_FPN_3x_deform.yaml'
 
 
 cfg = get_cfg()
@@ -28,7 +28,8 @@ cfg = get_cfg()
 cfg.merge_from_file(model_name)
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5  # set threshold for this model
 # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
-cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
+#cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
+cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50_deform.pth'
 
 if not torch.cuda.is_available():
     cfg.MODEL.DEVICE='cpu'