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

convert to tflite #3

Open
avber opened this issue Sep 22, 2022 · 3 comments
Open

convert to tflite #3

avber opened this issue Sep 22, 2022 · 3 comments

Comments

@avber
Copy link

avber commented Sep 22, 2022

Hi,

Is it possible to convert effdet_lite to the tflite format? If yes, could you add a code sample?

Thanks

@Burf
Copy link
Owner

Burf commented Sep 28, 2022

First, please refer to function of tfdet.export.tflite.
Usage will be updated soon.

@avber
Copy link
Author

avber commented Oct 26, 2022

Hi @Burf ,

I have EffdetLite4 and MobileVit models trained on similar but not identical datasets. MobileVit model achieved more than 94% accuracy.

Could you show an example of how to replace classification head with the trained MobileVit model and export a combined model to tflite ?

The MobileVit model will need the original input image, not the preprocessed one. max_output_size, iou__threshold and score_threshold for bbox predictions can be hardcoded.

The intention is to use the combined model in tfjs.

Thanks

@Burf
Copy link
Owner

Burf commented Mar 9, 2023

I don't know how the MobileVit structure or code is structured. Please refer to "usage/export/object_detection".

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