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

How to predict with batch size ? #35

Open
Vuong02011996 opened this issue Aug 22, 2021 · 1 comment
Open

How to predict with batch size ? #35

Vuong02011996 opened this issue Aug 22, 2021 · 1 comment

Comments

@Vuong02011996
Copy link

Vuong02011996 commented Aug 22, 2021

Thannks Vladimir Iglovikov! Great works, but I only predict with single image by using model.predict_jsons(image), how to use batch size in there?

@PVGorshenin
Copy link

@Vuong02011996 I assume you are using model of class Model, defined in retinaface.predict_single. predict_json is the inference just in case of one image.
Take a look on retinaface.inference. It is inference for the batch form. It has data preparation and loading under the hood. You have to pass arguments to script in the command line. It seems no explanation in the READ.me. But after taking a look on the code in inference.py things start to be clean.

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