Spaces:

danie94-lml
/

1Baja-Nest-segmentation

Sleeping

danie94-lml commited on Feb 10, 2024

Commit

22d6c0a

verified ·

1 Parent(s): 10db898

Update app.py

cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.70 change to 0.65

Files changed (1) hide show

app.py CHANGED Viewed

@@ -83,7 +83,7 @@ def app():
     cfg.MODEL.WEIGHTS = output_dir + "model_best.pth"
     cfg.MODEL.DEVICE = "cpu"
-    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.70
     predictor = DefaultPredictor(cfg)
     default_image_path = 'image/DJI_0142_frame_0041.jpg'

     cfg.MODEL.WEIGHTS = output_dir + "model_best.pth"
     cfg.MODEL.DEVICE = "cpu"
+    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.65
     predictor = DefaultPredictor(cfg)
     default_image_path = 'image/DJI_0142_frame_0041.jpg'