Spaces:

Humboldt
/

Trees

Runtime error

Humboldt commited on Oct 5, 2023

Commit

7a16ad9

1 Parent(s): 47882e0

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.1 #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=0.1, #FLAGS.yolo_score_threshold,
         soft_nms_sigma=0.5
     )