search

Syn-McJ / TFClassify-Unity

An example of using Tensorflow with Unity for image classification and object detection.
MIT License
167 stars 47 forks source link

about importing other pre-trained model #2

Closed tomtomtong closed 6 years ago

tomtomtong commented 6 years ago

Hello. How do you convert the pre-trained model into .bytes format? Do you use freeze_graph.py? Thx

Syn-McJ commented 6 years ago

Hi @tomtomtong,

No, I did not use freeze_graph, I just took the model from tensorflow repo and simply renamed it so that it has .bytes extension instead of .pb. Same thing for some MobileNet models I tried, for example the one I got flowing this article. Simply switch extension to .bytes and use it as you would use it in the android example from tensorflow repo.

Syn-McJ commented 6 years ago

I assume your question is answered, so I'm gonna close this issue. Feel free to reopen if anything is unclear.

tomtomtong commented 6 years ago

Thx a lot.