Spaces:

Noursine
/

Final-rooftop-detection

Sleeping

Noursine commited on Sep 10, 2025

Commit

4cdf8a4

verified ·

1 Parent(s): 2f89f4b

Update app4.py

Files changed (1) hide show

app4.py CHANGED Viewed

@@ -25,7 +25,7 @@ det_cfg = get_cfg()
 det_cfg.merge_from_file(
     model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
 )
-det_cfg.MODEL.WEIGHTS = "/app/model_final.pth"   # your trained weights
 det_cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 det_cfg.MODEL.DEVICE = "cpu"
 det_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1

 det_cfg.merge_from_file(
     model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
 )
+det_cfg.MODEL.WEIGHTS = "/app/model_final2.pth"   # your trained weights
 det_cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 det_cfg.MODEL.DEVICE = "cpu"
 det_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1