onnx / models

A collection of pre-trained, state-of-the-art models in the ONNX format
http://onnx.ai/models/
Apache License 2.0
7.46k stars 1.36k forks source link

Improve accuracy of the ultraface-rfb-640.onnx model #630

Open isaacporras opened 7 months ago

isaacporras commented 7 months ago

Ask a Question

Question

Hi,

Currently, I'm using version-RFB-640.onnx to perform face detection on a video from a cpp program. It works well on some videos but on others, it doesn't. The detection seems to be intermittent, at some frames, it detects some of the faces, but most of the time it doesn't. Is there a way to maybe adjust the tolerance/confidence of the model or improve the accuracy of the model?

Frame where onnx model did detect the face (the girl with purple blazer):

accurate_onnx

Frame where it didn't detect the face:

not_accurate_onnx

As you can see, it doesn't seem to be working on the more further faces or from the guys on the right.