fi-ber-detec-api

Sleeping

mosidi commited on Feb 24, 2023

Commit

1eb44a8

1 Parent(s): 53590df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,7 +91,7 @@ def inference(image_url, image, min_score):
         # Model expect BGR!
         im = image[:,:,::-1]
-    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
     predictor = DefaultPredictor(cfg)
     outputs = predictor(im)

         # Model expect BGR!
         im = image[:,:,::-1]
+    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.005
     predictor = DefaultPredictor(cfg)
     outputs = predictor(im)