Spaces:

SIH
/

building-segmentation

Runtime error

hlydecker commited on Nov 6, 2023

Commit

05d4fcf

•

1 Parent(s): c2432f1

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/buildingsv1_cfg.yaml")
 cfg.MODEL.DEVICE='cpu'
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
 cfg.MODEL.WEIGHTS = "model_weights/buildingsv1_best.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)

 cfg = get_cfg()
 cfg.merge_from_file("model_weights/buildingsv1_cfg.yaml")
 cfg.MODEL.DEVICE='cpu'
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.35
 cfg.MODEL.WEIGHTS = "model_weights/buildingsv1_best.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)