Spaces:

Humboldt
/

Trees

Runtime error

Humboldt commited on Oct 10, 2023

Commit

6a5ec6b

1 Parent(s): d6586ba

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -246,7 +246,7 @@ def yolo_nms(outputs, anchors, masks, classes):
         scores=scores,
         max_output_size=FLAGS.yolo_max_boxes,
         iou_threshold=FLAGS.yolo_iou_threshold,
-        score_threshold=0.01, #FLAGS.yolo_score_threshold,
         soft_nms_sigma=0.5
     )

         scores=scores,
         max_output_size=FLAGS.yolo_max_boxes,
         iou_threshold=FLAGS.yolo_iou_threshold,
+        score_threshold=FLAGS.yolo_score_threshold,
         soft_nms_sigma=0.5
     )