YOLO and OBB

kerrliam3

Hi there, I am looking for information on the support of Oriented Bounding Box (OBB) models for any YOLO version compatible with Luxonis. I haven't been able to find any relevant details could you clarify if OBB models are supported and, if so, how they can be integrated?

Thanks.

jakaskerl

kerrliam3
Hasn't yet been tested afaik
cc @JanCuhel to confirm and for tracking

Thanks,
Jaka

JanCuhel

Hi kerrliam3,

yes, unfortunately, we don't support YOLO OBB models at the moment. It is definitely planned to do so in the future, but we don't have an exact date yet.

If you need to use OBB models now, I'd suggest to you to convert the model to .blob, use NeuralNetwork node and process the output on host (check this and this). For converting a .pt model to an ONNX model file of YOLOv8 and 11, you can our tools-cli, though please be aware, that we slightly edited the OBB Head by pruning some operations, so these operations would need to be added to the post-processing on host.

Kind regards,
Jan