Spaces:

SIH
/

building-segmentation

Runtime error

hlydecker commited on Sep 28, 2023

Commit

2b1278f

•

1 Parent(s): 31c964b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 from detectron2.checkpoint import DetectionCheckpointer
 cfg = get_cfg()
-cfg.merge_from_file(model_zoo.get_config_file('COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml'))
 cfg.MODEL.DEVICE='cpu'
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"

 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"