Only save results with a given confidence percentage or above

[JaimyvS]

Search before asking

• I have searched the Pytorch-Wildlife issues and found no similar bug report.

Question

Is there currently a function in Pytorch Wildlife to only export results when they are above a given confidence level. Either for the JSON output or the cropped output. If not, I would love to put this in as a feature request.

Thanks!

Additional

No response

[zhmiao]

Hello @JaimyvS, currently we don't have a function that filter the outputs by confidence. It will be a good postprocessing function in addition to the current ones. We will be happy to add it in our future iterations!

[Wologman]

I would have prefered the original behaviour for the .json output, though I could see why others might want them filtered. Perhaps we could have this as an optional flag rather than a built-in behaviour?

For my purposes, we are still interested performing further analysis on the empty images. For example to calibrate our models against human verified examples, or to make decisions about appropriate thresholds. If the empty images (or images where the confidences are below the threshold) are not included in the .json files, then I need to handle it myself by adding them back in, which is OK, but extra complexity.