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

Where did the rknn model come from? #3

Open
ststare opened this issue Jan 5, 2024 · 8 comments
Open

Where did the rknn model come from? #3

ststare opened this issue Jan 5, 2024 · 8 comments

Comments

@ststare
Copy link

ststare commented Jan 5, 2024

Hi! Can you tell me where you got the rknn model from? (yolov5s-640-640.rknn) Or if you converted a .pt model, how was it converted and where did this .pt model come from?

@Qengineering
Copy link
Owner

Take a look here: https://github.com/airockchip/rknn_model_zoo

@ststare
Copy link
Author

ststare commented Jan 8, 2024

Thanks for the answer!

I have already become acquainted with the models that are in this repository. I followed the instructions described in the documents rknn toolkit2 and from yolov5s.onnx (this file is located here ) received yolov5s.rknn . Now when I open this received model and the model that you have (I open it in the Netron-7.3.6.AppImage application), they differ in outputs and, in general, completely internal architecture. Therefore, your model was definitely not obtained from rknn_model_zoo. And if you did use their conversion method, what .onnx model did you use initially?

Screenshot

On the left is your model, on the right is the model from rknn_model_zoo.

@Qengineering
Copy link
Owner

OK. This can also be a candidate. I'm also not sure after all this time.

@ststare
Copy link
Author

ststare commented Jan 8, 2024

Unfortunately, this model is also different. I understand that you did not convert the model yourself, but found a ready-made one. Did I understand correctly?

But then how is main.cpp written and what variables may need to be changed regarding which .rknn model we are using.

@Qengineering
Copy link
Owner

Indeed I found a ready-made one. Main.cpp is more or less a copy of the given example by Radxa

@ststare
Copy link
Author

ststare commented Jan 8, 2024

Thank you for your sincere answer! I would be very grateful if you could tell me in more detail about example by Radxa or maybe give me a link.

And maybe the last question, do you know which variables are responsible for the parameters of the .rknn model? So I can try to understand how to change them, relative to my model

@Qengineering
Copy link
Owner

Did you see this video from a colleague?

@ststare
Copy link
Author

ststare commented Jan 11, 2024

Thanks, I'll take a look!

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

No branches or pull requests

2 participants