Skip to content

Commit

Permalink
Update NMS threshold based on anecdotal changes after sigmoid fix (#1702
Browse files Browse the repository at this point in the history
)

0.9 is too high

Theres still no reasonable default, but lets try our best
  • Loading branch information
LukeWood authored Apr 11, 2023
1 parent eba5770 commit c142683
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ class MultiClassNonMaxSuppression(keras.layers.Layer):
IoU threshold for two boxes to be considered same for suppression.
Defaults to 0.5.
confidence_threshold: a float value in the range [0, 1]. All boxes with
confidence below this value will be discarded, defaults to 0.9.
confidence below this value will be discarded, defaults to 0.5.
max_detections: the maximum detections to consider after nms is applied. A
large number may trigger significant memory overhead, defaults to 100.
max_detections_per_class: the maximum detections to consider per class
Expand All @@ -45,7 +45,7 @@ def __init__(
bounding_box_format,
from_logits,
iou_threshold=0.5,
confidence_threshold=0.9,
confidence_threshold=0.5,
max_detections=100,
max_detections_per_class=100,
**kwargs,
Expand Down

0 comments on commit c142683

Please sign in to comment.