Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,13 +25,13 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
|
|
25 |
from detectron2.checkpoint import DetectionCheckpointer
|
26 |
|
27 |
cfg = get_cfg()
|
28 |
-
cfg.merge_from_file("model_weights/
|
29 |
cfg.MODEL.DEVICE='cpu'
|
30 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
|
31 |
#cfg.MODEL.WEIGHTS = "model_weights/model_final.pth"
|
32 |
#cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
|
33 |
-
cfg.MODEL.WEIGHTS = "model_weights/
|
34 |
-
cfg.MODEL.ROI_HEADS.NUM_CLASSES =
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
|
37 |
def segment(im):
|
|
|
25 |
from detectron2.checkpoint import DetectionCheckpointer
|
26 |
|
27 |
cfg = get_cfg()
|
28 |
+
cfg.merge_from_file("model_weights/gsu-lcz-sample-model_cfg.yaml")
|
29 |
cfg.MODEL.DEVICE='cpu'
|
30 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
|
31 |
#cfg.MODEL.WEIGHTS = "model_weights/model_final.pth"
|
32 |
#cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
|
33 |
+
cfg.MODEL.WEIGHTS = "model_weights/gsu-lcz-sample-model_best.pth"
|
34 |
+
cfg.MODEL.ROI_HEADS.NUM_CLASSES = 13
|
35 |
predictor = DefaultPredictor(cfg)
|
36 |
|
37 |
def segment(im):
|