0

I would like to ask some help. I've been struggling about how to convert Yolov5 model to Tensorflow lite. Any suggestion guys, how can I convert my Yolov5 model to tensorflow lite?

Mike B
  • 2,136
  • 2
  • 12
  • 31
exiled_coder
  • 1
  • 1
  • Please edit the question to limit it to a specific problem with enough detail to identify an adequate answer. – Community Mar 13 '22 at 15:30
  • Hi! You can use TFLite converter API for conversion. Attaching a similar thread for reference. https://stackoverflow.com/questions/70670841/how-to-convert-yolov5-trained-model-into-tflite-in-google-colab-to-install-on-mo –  Apr 04 '22 at 09:53
  • Add code to your question so that we can understand what you have tried – Mike B Jul 21 '22 at 12:50

1 Answers1

2

As sepecified in the official Yolov5 export.py (https://github.com/ultralytics/yolov5/blob/master/export.py) you should run:

python path/to/export.py --weights yolov5s.pt --include tflite
Mike B
  • 2,136
  • 2
  • 12
  • 31