File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
keras_cv/layers/object_detection Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments