e2e-tflite-tutorials icon indicating copy to clipboard operation
e2e-tflite-tutorials copied to clipboard

Published 20 hours ago • verified publisher icon ml-gde

YOLO: create and publish a tflite model on TF Hub

Open margaretmz opened this issue 4 years ago • 2 comments

margaretmz avatar Jul 17 '20 02:07 margaretmz

@hunglc007 who created this repo is working on converting YOLOv3_tiny to TF Lite. So far he has succeeded on creating a YOLOv3_tiny model that can run on TF Lite GPU.

The model training part is still needed to be done in darknet. Hung is working on implementing a training pipeline on TF2.

khanhlvg avatar Jul 20 '20 02:07 khanhlvg

@hunglc007 could you provide an update on this project?

If you were able to obtain a TFLite variant of YOLOv3_tiny could you help me with the following?

  • Notebook that shows the conversion process and inference in Python
  • Links to the TFLite models

Additionally, would you be interested in publishing these models from TensorFlow Hub? That way the models could reach an even bigger audience.

sayakpaul avatar Dec 06 '20 03:12 sayakpaul