search

ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
49.5k stars 16.08k forks source link

Tflite convertion #369

Closed e-shawakri closed 4 years ago

e-shawakri commented 4 years ago

Any one found a way to convert the yolov5l into tflite?

glenn-jocher commented 4 years ago

The default pipeline is through ONNX. https://github.com/onnx/onnx-tensorflow

e-shawakri commented 4 years ago

it's not working through ONNX, when the model converting from ONNX into pb it removes the input and the output arrays which are named "images" and "output" and therefore the model can't be converted into tflite, because the input and output arrays are vanished when converting the model from ONNX into .pb.

glenn-jocher commented 4 years ago

@e-shawakri sounds like an issue for the onnx-tensorflow repo rather than this repo.

e-shawakri commented 4 years ago

@glenn-jocher Thanks for your replay, but i'm not asking about the problem. I'm asking if anybody do the pipeline process and succeed to convert the model into tflite model.

github-actions[bot] commented 4 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.