lkeab commited on
Commit
900e3ff
1 Parent(s): aa1da84

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,10 +34,10 @@ model_name='./configs/COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yam
34
 
35
  cfg = get_cfg()
36
  # add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
37
- cfg.merge_from_file(model_zoo.get_config_file(model_name))
38
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model
39
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
40
- cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(model_name)
41
 
42
  if not torch.cuda.is_available():
43
  cfg.MODEL.DEVICE='cpu'
 
34
 
35
  cfg = get_cfg()
36
  # add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
37
+ cfg.merge_from_file(model_name)
38
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model
39
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
40
+ cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
41
 
42
  if not torch.cuda.is_available():
43
  cfg.MODEL.DEVICE='cpu'