jetson-inference icon indicating copy to clipboard operation
jetson-inference copied to clipboard

Modifications to detectNet.cpp for use with Yolo NAS or Yolov8

Open lars-nagel opened this issue 1 year ago • 1 comments

Since the newer Yolo architectures are not officially supported for object detection: I wonder if anyone from the community successfully modified the pre- and post processing in detectNet.cpp to use it with Yolo NAS or Yolov8 and is willing to share how they did it? I would love to use those with jetson-inference, which is a great framework per se. Any help is appreciated!

lars-nagel avatar Feb 22 '24 22:02 lars-nagel

I put some time into it. I think I was able to modify the onnx models, to have the outputs compatible with detectNet. But since I am using the JetsonNano Dev Kit (not Orin) the YOLO models just can't work due to unsupported operations within the model.

If that's not your issue and you just need to transform the model, this post might help you https://vilsonrodrigues.medium.com/add-non-maximum-suppression-nms-to-object-detection-model-using-onnx-7639a698cf05

pq53ui avatar Mar 06 '24 16:03 pq53ui