openvinotoolkit / open_model_zoo

Pre-trained Deep Learning models and demos (high quality and extremely fast)
https://docs.openvino.ai/latest/model_zoo.html
Apache License 2.0
4.1k stars 1.37k forks source link

Use batch size bigger than 1 for head-pose-model #3637

Closed vadik6666 closed 1 year ago

vadik6666 commented 1 year ago

Hi, is it possible to use head-pose-estimation-adas-001 model from model-zoo with a batch size 2 or bigger? In the model page the input described as a 1, 3, 60, 60 Doing simply model([input]), where input is e.g. 2, 3, 60, 60 gives an error RuntimeError: The input blob size is not equal to the network input size: got 21600 expecting 10800 Maybe there is way? Somehow convert to openvino with batch dimension not fixed, but to do this I need original model, am correct?

eaidova commented 1 year ago

@vadik6666 you can reshape openvino model for specific batch size (even dynamic), you can find tutorial hows to do it here: https://github.com/openvinotoolkit/openvino_notebooks/blob/main/notebooks/002-openvino-api/002-openvino-api.ipynb

there is no need in original model

vadik6666 commented 1 year ago

Thank you @eaidova