Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
|
|
25 |
from detectron2.checkpoint import DetectionCheckpointer
|
26 |
|
27 |
cfg = get_cfg()
|
28 |
-
cfg.merge_from_file("model_weights/treev1_cfg.
|
29 |
cfg.MODEL.DEVICE='cpu'
|
30 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
|
31 |
cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"
|
|
|
25 |
from detectron2.checkpoint import DetectionCheckpointer
|
26 |
|
27 |
cfg = get_cfg()
|
28 |
+
cfg.merge_from_file("model_weights/treev1_cfg.yaml")
|
29 |
cfg.MODEL.DEVICE='cpu'
|
30 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
|
31 |
cfg.MODEL.WEIGHTS = "model_weights/treev1_best.pth"
|