search

mystic123 / tensorflow-yolo-v3

Implementation of YOLO v3 object detector in Tensorflow (TF-Slim)
https://medium.com/@pawekapica_31302/implementing-yolo-v3-in-tensorflow-tf-slim-c3c55ff59dbe
Apache License 2.0
893 stars 353 forks source link

Convert .pb to .tflite file #89

Closed Bao150297 closed 4 years ago

Bao150297 commented 4 years ago

I'm facing with running yolov3-tiny on mobile device. Can anyone show me how to convert .pb file to .tflite so I can use it with tensorflow-lite? Thanks in advance

peace195 commented 4 years ago

@DoriHp I did it here: https://github.com/peace195/tensorflow-lite-yolo-v3

I would I would appreciate if you give me a star for this project 👍

Bao150297 commented 4 years ago

@peace195 I found your repo and tried both it and built-in tflite converter of tensorflow. Both worked well. I also gave you a star for your effort. The problem I'm facing now is implement it on android device. I'm not a professional in machine learning so it will be very helpful if you can give me a suggestion to resolve it.

nscotto commented 4 years ago

I have just tried it as well, conversion worked well, another star for you! ;) Same as DoriHP, I think some details on how to recover the probabilities and the boxes would be really appreciated. Thank you.

peace195 commented 4 years ago

@nscotto Thanks! I will update it when I have time.

peace195 commented 4 years ago

@nscotto @DoriHp Here is the example of tflite usage. https://github.com/peace195/tensorflow-lite-yolo-v3/blob/master/tflite_example.py

It is in python code. I will update Android and iOS version later.

Also, I need some helps: https://github.com/peace195/tensorflow-lite-yolo-v3/issues/1