aerial-segmentation-model-selection

Runtime error

hlydecker commited on Oct 30, 2023

Commit

2a31c27

1 Parent(s): c2f4d83

patch: fix thresh

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/tree_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.25
 cfg.MODEL.WEIGHTS = "model_weights/tree_model.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 predictor = DefaultPredictor(cfg)

 cfg = get_cfg()
 cfg.merge_from_file("model_weights/tree_cfg.yml")
 cfg.MODEL.DEVICE='cpu'
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
 cfg.MODEL.WEIGHTS = "model_weights/tree_model.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 predictor = DefaultPredictor(cfg)