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

Thankyou for excllent work. I am able to run demo_client but please i want to know how i can select the model of my choice or put custom trained model. #122

Open
ZubairKhan001 opened this issue Dec 17, 2023 · 1 comment

Comments

@ZubairKhan001
Copy link

No description provided.

@SthPhoenix
Copy link
Owner

SthPhoenix commented Dec 19, 2023

To select different models from supported list you should just replace following lines in compose/.env:

DET_NAME=yolov5l-face
REC_NAME=w600k_r50

Adding custom detector is a bit tricky since you need to implement postprocessing specific for model, while face recognition models based on official InsightFace arcface architecture can be just placed in folder models/onnx/custom_rec_model.

If you need to add additional compatible rec models they should be declared in https://github.com/SthPhoenix/InsightFace-REST/blob/master/src/api_trt/modules/configs.py
You can use custom_rec_model config as reference

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