YOLOX OpenVINO Model Batch Inference issue

[LiuYiShan613]

I am currently conducting batch inference on the YOLOX model using the OpenVINO IR format. When the batch size is set to 1, the IR model's output shape is (1, 3549, 85). However, when the batch size is set to 'n', the output shape becomes (1, 3549 * n, 85). I anticipate the output shape to be (n, 3549, 85) to align with the requirements of the demo_postprocess function.

Consequently, I attempt to reshape the IR model's output from (1, 3549 * n, 85) to (n, 3549, 85). Unfortunately, the bounding boxes are not predicted correctly. How should I process this output data to ensure it fits the demo_postprocess function?