Skip to content
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

Default parameter values #26

Open
woctezuma opened this issue Dec 4, 2020 · 1 comment
Open

Default parameter values #26

woctezuma opened this issue Dec 4, 2020 · 1 comment

Comments

@woctezuma
Copy link

woctezuma commented Dec 4, 2020

Hello,

Thank you for sharing this work as a PyPI package!

I have questions about default parameter values.

arg("-m", "--max_size", type=int, help="Resize the largest side to this number", default=960)
arg("-b", "--batch_size", type=int, help="batch_size", default=1)
arg("-j", "--num_workers", type=int, help="num_workers", default=12)
arg("--confidence_threshold", default=0.7, type=float, help="confidence_threshold")
arg("--nms_threshold", default=0.4, type=float, help="nms_threshold")

Are these values recommended based on the training dataset and procedure of RetinaFace?
Or are these values which worked best for you?

More specifically, I don't know which value for max_size to choose for my personal dataset of mixed photos and hand-drawn/computer-assisted artworks. I am analysing 300x450 images. So, maybe I should upsample them 2x, and go for 900 max size.

As for the confidence threshold, I would settle for something closer to 0.5 (or even less, to detect hand-drawn faces), and lower the NMS threshold as well (to avoid detecting too many boxes). I understand that it might not be the case for people using the detector on natural photography. Do you think it is a bad idea to lower the threshold?

@woctezuma
Copy link
Author

Apparently, training data was 840x840 resolution.

image_size: [840, 840]

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant