You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
Currently, the object detection widgets don't have a threshold parameter, so they just use the hardcoded one from the object detection pipeline, which is 0.9 as seen here.
However, with more and more object detection models coming, we definitely need an adjustable threshold button for the widget, as for instance now Deformable DETR checkpoints, which need a lower threshold to detect objects in a scene, simply say "no objects detected":
Describe the solution you'd like
Adding a "threshold button" to the inference widgets. Perhaps make it possible to set a default threshold in the model card.
The text was updated successfully, but these errors were encountered:
Hey @osanseviero, threshold is already supported in the pipeline class, I'm updating the pipeline to use the post_process_object_detection methods instead of post_process with this PR. We still need to add a threshold slider to the inference widgets though.
Is your feature request related to a problem? Please describe.
Currently, the object detection widgets don't have a
threshold
parameter, so they just use the hardcoded one from the object detection pipeline, which is 0.9 as seen here.However, with more and more object detection models coming, we definitely need an adjustable threshold button for the widget, as for instance now Deformable DETR checkpoints, which need a lower threshold to detect objects in a scene, simply say "no objects detected":
Describe the solution you'd like
Adding a "threshold button" to the inference widgets. Perhaps make it possible to set a default threshold in the model card.
The text was updated successfully, but these errors were encountered: