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

Does convert_model_to_tm support tensorflow h5 file? #297

Open
shiyileiecnu opened this issue May 12, 2020 · 1 comment
Open

Does convert_model_to_tm support tensorflow h5 file? #297

shiyileiecnu opened this issue May 12, 2020 · 1 comment

Comments

@shiyileiecnu
Copy link

TensorFLow can save the model as h5 format.
I used convert_model_to_tm to transfer tensorflow h5 file to tengine file with no failure as below
./install/convert_tools/convert_model_to_tm -f tensorflow -m ./my_model.h5 -o ./my_model.tmfile
Create tengine model file done: ./my_model.tmfile

but the final tmfile size is very small ( only "228" as below)
228 17:53 ./my_model.tmfile

When I load the tmfile in application, it shows
"Model name : ../my_model.tmfile
tengine model file : ../my_model.tmfile
label file : ./synset_words.txt
image file : ./dog.jpg
img_h, imag_w, scale, mean[3] : 224 224 0.017 104.007 116.669 122.679
Cannot find tensor node_index: 0, tensor_index: 0"

I just want to know whether convert_model_to_tm support tensorflow h5 file

@liqi-c
Copy link
Collaborator

liqi-c commented May 18, 2020

we only support .pb file format for tensorflow .
you can transform h5 to .pb first . thanks .

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