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

Could I re-train it with my own datasets? #7

Open
haophancs opened this issue Sep 11, 2018 · 4 comments
Open

Could I re-train it with my own datasets? #7

haophancs opened this issue Sep 11, 2018 · 4 comments

Comments

@haophancs
Copy link

No description provided.

@ofirsteinherz
Copy link

Hi,
Yes you definitely can!
If the dataset is labeled, you should complete tensorflow for poets. Instead of download the training images from google, add yours.

Then, you have labels and NN (not tf-lite). So in tensorflow for poets 2 tflite you will optimize your model using the TFLite converter.

After that, just swich mobilenet_quant_v1_224.tflite and labels.txt with the files that tf for poets executed.

Good luck :-)

@bathanfritz
Copy link

I tried to retrain it with my own data set and used mobilenet_1.0_224 as tfhub module but it crashes, please help me

@soum-io
Copy link
Contributor

soum-io commented Feb 5, 2019

My recent PR has added support for float models, which is probably the type of model you trained. Simply change the variable QUANT to false in TensorFlowImageClassifier.java along with changing the model and lebels name.

@sai-pher
Copy link

Hi there.
I wanted to know if you retrain with a new dataset, how do you order the new labels in the label.txt file?
Does the order matter? And if so, how can one be sure of the order to use?

I working with a new model built using tesnorflow 1.13.1's Keras.

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

5 participants