Hugging Face's logo

Spaces:
SIH
/
tree-segmentation
Runtime error

App Files Community
hlydecker commited on
Commit
a210f65
·
1 Parent(s): 6955470

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ from detectron2.checkpoint import DetectionCheckpointer
27
  cfg = get_cfg()
28
  cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
- cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
31
  cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"
32
  cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
33
  predictor = DefaultPredictor(cfg)
 
27
  cfg = get_cfg()
28
  cfg.merge_from_file("model_weights/buildings_poc_cfg.yml")
29
  cfg.MODEL.DEVICE='cpu'
30
+ cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
31
  cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"
32
  cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
33
  predictor = DefaultPredictor(cfg)