aerial-segmentation-model-selection

Runtime error

hlydecker commited on Sep 28, 2023

Commit

1e15a0a

1 Parent(s): 895a2c1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ 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)

 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/model_final.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)