Only save results with a given confidence percentage or above

microsoft / CameraTraps

PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.

https://cameratraps.readthedocs.io/en/latest/

MIT License

722 stars 234 forks source link

Only save results with a given confidence percentage or above #496

Open JaimyvS opened 2 months ago

JaimyvS commented 2 months ago

Search before asking

[X] 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 commented 2 months ago

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!