how to doInference(**) on OBB(oriented Bounding box)

[minwim]

if I use yolov5's obb model to .engine, How i to modify doInference function. I want to get (x,y,w,h, theta) not HBB(x,y,w,h)