Skip to content

Commit ec39676

Browse files
authored
Update multi_class_non_max_suppression.py (#1318)
* Update multi_class_non_max_suppression.py This will give much better results aesthetically to our users. * Update multi_class_non_max_suppression.py
1 parent 1ad0f74 commit ec39676

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras_cv/layers/object_detection/multi_class_non_max_suppression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class MultiClassNonMaxSuppression(tf.keras.layers.Layer):
3131
IoU threshold for two boxes to be considered same for suppression. Defaults
3232
to 0.5.
3333
confidence_threshold: a float value in the range [0, 1]. All boxes with
34-
confidence below this value will be discarded. Defaults to 0.05.
34+
confidence below this value will be discarded. Defaults to 0.9.
3535
max_detections: the maximum detections to consider after nms is applied. A large
3636
number may trigger significant memory overhead. Defaults to 100.
3737
max_detections_per_class: the maximum detections to consider per class after
@@ -43,7 +43,7 @@ def __init__(
4343
bounding_box_format,
4444
from_logits,
4545
iou_threshold=0.5,
46-
confidence_threshold=0.5,
46+
confidence_threshold=0.9,
4747
max_detections=100,
4848
max_detections_per_class=100,
4949
**kwargs,

0 commit comments

Comments
 (0)