Spaces:

SIH
/

building-segmentation

Runtime error

hlydecker commited on Sep 28, 2023

Commit

a210f65

•

1 Parent(s): 6955470

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ from detectron2.checkpoint import DetectionCheckpointer
 cfg = get_cfg()
 cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)

 cfg = get_cfg()
 cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
 cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)