lkeab commited on
Commit
e761b96
1 Parent(s): ffb9fb5

update app

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. output_3x_transfiner_r101_deform.pth +3 -0
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/transfiner/mask_rcnn_R_50_FPN_3x_deform.yaml'
24
 
25
 
26
  cfg = get_cfg()
@@ -30,7 +30,7 @@ cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model
30
  cfg.VIS_PERIOD = 100
31
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
32
  #cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
33
- cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50_deform.pth'
34
 
35
  if not torch.cuda.is_available():
36
  cfg.MODEL.DEVICE='cpu'
20
  from detectron2.data import MetadataCatalog
21
 
22
 
23
+ model_name='./configs/transfiner/mask_rcnn_R_101_FPN_3x_deform.yaml'
24
 
25
 
26
  cfg = get_cfg()
30
  cfg.VIS_PERIOD = 100
31
  # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as w ell
32
  #cfg.MODEL.WEIGHTS = './output_3x_transfiner_r50.pth'
33
+ cfg.MODEL.WEIGHTS = './output_3x_transfiner_r101_deform.pth'
34
 
35
  if not torch.cuda.is_available():
36
  cfg.MODEL.DEVICE='cpu'
output_3x_transfiner_r101_deform.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d5fefa123e70f379af846c979139e53e89b8c661b7285dd485e640fa86faead
3
+ size 294703133