Add YOLO11 Model Support

[lakshanthad]

Hello @marcoslucianops,

Great work on DeepStream-Yolo!

With the release of the latest Ultralytics YOLO11 models, it would be great to have DeepStream support for these models.

Thank you.

[neilyoung]

+1

[neilyoung]

@lakshanthad Oh, I see that the V8 converter also works for yolo11 models

[githubwalt]

What do you mean by "V8 converter"?

[neilyoung]

python3 utils/export_yoloV8.py -w yolo11s.pt

use the resulting onnx model then

[lakshanthad]

Great @neilyoung. However, it would be nice to have a separate section for YOLO11.

Btw, please refer to the below instructions for YOLOv8. Still waiting for merge:
#551

[neilyoung]

@lakshanthad Thanks for making me aware of this.

[worstshit]

python3 utils/export_yoloV8.py -w yolo11s.pt

use the resulting onnx model then

I tried this but it doesn't work. It will come into an AttributeError:
Can't get attribute 'C3k2' on <module 'ultralytics.nn.modules.block' from ....>
Can you please help tell how to fix this problem? Thank you.

[neilyoung]

@worstshit Well, I downloaded the weights from here and ran the script as shown

wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s.pt

[jugal-sheth]

@worstshit Kindly install updated ultralytics[export] package

pip3 install ultralytics[export]
pip3 install cmake
pip3 install onnxsim

[worstshit]

@worstshit Well, I downloaded the weights from here and ran the script as shown

wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s.pt

@neilyoung I'm sorry to ask again, i still failed with this export yolov11 model to onnx. I use jetson nx 16GB and tried as follows:
first, run the command pip3 install ultralytics[export], pip3 install cmake, pip3 install onnxsim.
second, git clone https://github.com/marcoslucianops/DeepStream-Yolo
cd DeepStream-Yolo
thirdly, wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s.pt
finally, python3 utils/export_yoloV8.py -w yolo11n.pt
and I come into the same error:
Can't get attribute 'C3k2' on <module 'ultralytics.nn.modules.block' from ....>
Can you please help tell is there missing some steps or something wrong? thank you.

[neilyoung]

Launch a clean Ubuntu 22.04 VM, install this repo and deps there and do the conversion