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

bbox, landmark = self.model.detect(cons_img, threshold=0.1, scale=1.0) problem #2627

Open
addiajsjjs opened this issue Aug 9, 2024 · 1 comment

Comments

@addiajsjjs
Copy link

self.model = insightface.model_zoo.get_model('retinaface_r50_v1') # ./.insightface
self.model.prepare(ctx_id=self.gpu_id, nms=0.4)
bbox, landmark = self.model.detect(cons_img, threshold=0.1, scale=1.0)

I want to use this code to perform detection, but the latest version of the insightface library has changed the detect function from its previous implementation.
image
Here is the latest information.
I found that the detect function in detection/retinaface/retinaface.py is the one I want to use. How should I use the detect function from that file?

@nttstar
Copy link
Collaborator

nttstar commented Aug 14, 2024

change to load the face-detection model under 'antelopev2' or 'buffalo_l' model package.

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

2 participants