-
Notifications
You must be signed in to change notification settings - Fork 107
[Face attributes Model] Output of sunglasses probability is always 1 #179
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Comments
Hi @chouxscream, thanks for the question! I've asked the team that worked on this model and they have some follow up questions: Yes, you are handling the sunglasses result correctly, and there's no need to perform softmax. Though we'll need more information to pinpoint the exact problem. If you could provide some failure cases, it would be easier for us to identify the root cause. Are you using RGB data? Regarding preprocessing, for optimal results, we perform an affine transformation to the images and crop the faces according to the following criteria: center of eyes and mouth are supposed to be well aligned horizontally and the distance from the top of the image to the eyes, from the eyes to the mouth, and from the mouth to the bottom should all equal to 1/3 of the image height. |
Hi chouxscream, thank you for the additional details. You're right it does appear that the results are incorrect and significantly differ from our previous runs, both online and offline. We'll need more information to dig into the root cause. Could you provide us with the following:
In the meantime, we'd recommend trying the quantized version as the accuracy should improve, depending on the platform you're deploying the model on. We also wanted to share that we'll soon be updating a new version trained using RGB instead of NIR data, so you should expect better accuracy in the near future release. |
I plan to export onnx model to other NPU device and since quantization is applied separately, I did not use quantized version from qualcomm.
So should I use NIR image input instead of RGB?
For preprocessing, crop face bounding box from scrfd-10g detector and resize to (128,128) |
Hello, I have question on face attributes model output.
While other outputs seems to be correct after using softmax function, output of sunglasses is always equal to or almost 1.0.
I have not used softmax to sunglasses, as pointed out in #137.
Regardless of input images with or without sunglasses on, the output does not represent sunglasses correctly.
Am I missing something, such as preprocessing input data? Below is the command that I have used.
python -m qai_hub_models.models.face_attrib_net.demo --image {image_path}
Thank you in advance.
The text was updated successfully, but these errors were encountered: