retinaface icon indicating copy to clipboard operation
retinaface copied to clipboard

How to predict with batch size ?

Open Vuong02011996 opened this issue 3 years ago • 1 comments

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?

Vuong02011996 avatar Aug 22 '21 09:08 Vuong02011996

@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.

PVGorshenin avatar Nov 09 '21 08:11 PVGorshenin