Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ from detectron2.utils.visualizer import ColorMode
|
|
23 |
model_path = 'model_final.pth'
|
24 |
|
25 |
cfg = get_cfg()
|
26 |
-
cfg.merge_from_file("
|
27 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
|
28 |
cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
|
29 |
cfg.MODEL.WEIGHTS = model_path
|
|
|
23 |
model_path = 'model_final.pth'
|
24 |
|
25 |
cfg = get_cfg()
|
26 |
+
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
|
27 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
|
28 |
cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
|
29 |
cfg.MODEL.WEIGHTS = model_path
|