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

[Request] Support for Tensorflow Lite #134

Open
NiklasWilson opened this issue May 19, 2020 · 7 comments
Open

[Request] Support for Tensorflow Lite #134

NiklasWilson opened this issue May 19, 2020 · 7 comments
Labels
enhancement New feature or request wontfix This will not be worked on

Comments

@NiklasWilson
Copy link

It would be awesome if the project had support for an additional step. Converting the trained modal to TFLite. TFlite modals are needed for things like the google coral.

@bertelschmitt
Copy link

Train_YOLO has the command line option "--is_tiny", which is said to "Use the tiny Yolo version for better performance and less accuracy. " Haven't checked whether it uses TFLite

@NiklasWilson
Copy link
Author

Thanks but I already use the tiny yolo. It works way better on a PI. ~250ms
The way TFLite works is that you First train a normal TF modal and than afterwards you convert it to a TFLite modal.

One could choose to use either the normal yolo or the tiny yolo which both result in a normal TF modal.

@bertelschmitt
Copy link

What is your experience with the Coral? I have one sitting around, but I didn't play much with it.

@NiklasWilson
Copy link
Author

NiklasWilson commented May 20, 2020

limited. I successfully used it with a pre-trained TFLite modal on my live camera feeds. It takes 30-60ms. That is where my experience ends with it.
After I have a way to train my own TFLite Modal I intend to play with it more.
But I actually haven't done much in the last couple months related to this been busy with other projects. From what I remember the process of converting a normal modal to a lite modal had a bit of learning curve, no easy tool out there when I looked.

@bertelschmitt
Copy link

That was pretty much my experience. My idea was to give cheap GPU-like hardware support to VMs, but I quickly got dissuaded ...

@robisen1
Copy link

robisen1 commented Sep 7, 2020

Train_YOLO has the command line option "--is_tiny", which is said to "Use the tiny Yolo version for better performance and less accuracy. " Haven't checked whether it uses TFLite

it uses tiny yolo. tiny yolo is much faster than a full set of weights but you can expect at least 1/3 less accuracy. tflite, in my experience, is faster that tiny yolo

@NiklasWilson
Copy link
Author

but you can expect at least 1/3 less accuracy. tflite, in my experience, is faster that tiny yolo

My understanding is Tiny yolo and tflite can both be used at the same.
Tiny yolo seemed accurate enough for my use cases. I think the minimum amount of training data for "accurate" results is higher on tiny yolo.
Tflite (non yolo) takes ~60ms on my pi
TinyYolo takes ~500ms on my pi
Yolo takes 1-2sec on my pi

@AntonMu AntonMu added wontfix This will not be worked on enhancement New feature or request labels Apr 25, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request wontfix This will not be worked on
Projects
None yet
Development

No branches or pull requests

4 participants