Spaces:
Runtime error
Runtime error
lkeab
commited on
Commit
•
e761b96
1
Parent(s):
ffb9fb5
update app
Browse files- app.py +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/
|
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 = './
|
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
|