tfjs icon indicating copy to clipboard operation
tfjs copied to clipboard

Model Face Detection Confidence Score Output or Minimum Confidence Score Threshold

Open thaddmt opened this issue 1 year ago • 0 comments

Please make sure that this is a feature request. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:feature_template

System information

  • TensorFlow.js version (you are using): @tensorflow-models/[email protected]
  • Are you willing to contribute it (Yes/No): Yes

Describe the feature and the current behavior/state. Currently the TFJS Face detection model doesn't allow for configuring the minimum confidence score threshold while this appears to be possible in MediaPipe itself. Previously it seems like the TFJS blazeface model returned a probability field.

Is it possible to add either of these so that we could filter out faces based on their confidence threshold or probability?

This other relevant issue https://github.com/tensorflow/tensorflow/issues/62683 mentions that the media pipe short range documentation shows that a detection confidence score should be returned.

Will this change the current api? How? Would need to add some configuration to set the minimum thresholds or change the output to return a probability.

Who will benefit with this feature? Users of TFJS face detection

Any Other info.

thaddmt avatar May 07 '24 19:05 thaddmt