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

custom model #71

Open
ghost opened this issue Feb 21, 2024 · 10 comments
Open

custom model #71

ghost opened this issue Feb 21, 2024 · 10 comments

Comments

@ghost
Copy link

ghost commented Feb 21, 2024

I want to use my own yolo model it is for sign language detection..
I tried to just replace the model and labels paths but nothing appear.. No errors and also no label or bounding box.
I don't know if there is something else needs to be changed
This how I change the code:
image
here is what is the terminal keep showing:
image

@abdelaziz-mahdy
Copy link
Owner

Your model didn't detect any objects, the before processing part is all of the objects detected

Check your weights and the way you exported your model

@ghost
Copy link
Author

ghost commented Feb 21, 2024

Your model didn't detect any objects, the before processing part is all of the objects detected

Check your weights and the way you exported your model

How can i train the model? I have the imgs with their annotations and apply this python code to get .pt file
image
after that I convert best.pt into .torchscript using the line you provided
!yolo mode=export model="your model" format=torchscript optimize

@abdelaziz-mahdy
Copy link
Owner

i dont see a problem with your code, can you try another model or other images

@ghost
Copy link
Author

ghost commented Feb 25, 2024

i dont see a problem with your code, can you try another model or other images

I have tried another model and its worked as expected, but the real time detection is too slow is their any way to improve it?

@abdelaziz-mahdy
Copy link
Owner

Sadly for now there is none, it's a limitation from pytorch it self, for the best speeds you will have to use tflite

@ghost
Copy link
Author

ghost commented Feb 25, 2024

Sadly for now there is none, it's a limitation from pytorch it self, for the best speeds you will have to use tflite

I already tried it but it requires flutter downgrade ,because it is not work with the latest flutter version 😢

@abdelaziz-mahdy
Copy link
Owner

abdelaziz-mahdy commented Feb 25, 2024

Sadly the package is using the latest pytorch release for mobile and no GPU support yet😢

@nqtworking
Copy link

Hi @zezo357 , I'm a beginner in computer vision and Flutter. I'm wondering if your pre-made example has the ability to convert images from the camera to int8 format and then pass them to a YOLO model exported as int8 to improve processing speed?

@abdelaziz-mahdy
Copy link
Owner

The example does convert camera image to it's float32 format since that's the input for yolo yes

@nqtworking
Copy link

@zezo357 thanks for your feedback!

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