Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YOLOv11 Model Naming Support #1761

Open
Lunar-Academy opened this issue Feb 3, 2025 · 4 comments
Open

YOLOv11 Model Naming Support #1761

Lunar-Academy opened this issue Feb 3, 2025 · 4 comments
Labels
enhancement New feature or request

Comments

@Lunar-Academy
Copy link

From my testing, creating a YOLOv11 model and quantizing/converting to rknn works with PhotonVision's object detection. However, the model must be named as a YOLOv8 model. From looking at the RKNN toolkit-2 examples (RKNN Model Zoo), the way of running the models seem identical. I wasn't able to find any differences in the files (YOLOv8 and YOLOv11) used for processing the models. Because of this, PhotonVision's backend for YOLOv8 detection works for YOLOv11.

Model I have been testing with (Github issues doesn't let me upload the file directly, so it's a link to my team's code.)

@Lunar-Academy Lunar-Academy added the enhancement New feature or request label Feb 3, 2025
@Sam948-byte
Copy link
Contributor

This is currently in progress, I'm working on integrating this into the JNI right now.

@Lunar-Academy
Copy link
Author

While you're doing that, do you think you would be able to modify the npu core selection? From what I can tell, PV currently relies on autoselecting the npu cores.

RKNN Object Detector file

If the -1 on core selection parameter of the create method were be changed to 210, theoretically PV would be able to effectively utilize all npu cores.

@Sam948-byte
Copy link
Contributor

Sam948-byte commented Feb 4, 2025

I believe that's currently addressed in another PR. Not quite sure of the number. #1744

@Lunar-Academy
Copy link
Author

Ah, got it

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants