lkeab commited on
Commit
bb15987
1 Parent(s): b65bf73

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -20,7 +20,7 @@ from detectron2.utils.visualizer import Visualizer
20
  from detectron2.data import MetadataCatalog
21
 
22
 
23
- model_name='./configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x_4gpu_transfiner.yaml'
24
 
25
 
26
  cfg = get_cfg()
@@ -28,7 +28,8 @@ cfg = get_cfg()
28
  cfg.merge_from_file(model_name)
29
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model
30
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
31
- cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
 
32
 
33
  if not torch.cuda.is_available():
34
  cfg.MODEL.DEVICE='cpu'
 
20
  from detectron2.data import MetadataCatalog
21
 
22
 
23
+ model_name='./configs/transfiner/mask_rcnn_R_50_FPN_3x_deform.yaml'
24
 
25
 
26
  cfg = get_cfg()
 
28
  cfg.merge_from_file(model_name)
29
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model
30
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
31
+ #cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
32
+ cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50_deform.pth'
33
 
34
  if not torch.cuda.is_available():
35
  cfg.MODEL.DEVICE='cpu'