如何在 google colab 中将 yolov5 训练模型转换为 tflite 以安装在移动应用程序上？

Question

我已经使用 Yolov5 在 google colab 中训练模型，如何导出训练模型的 .tflite 文件以便它可以用于移动应用程序。谢谢你帮我

Answer 1

假设您拥有最新的 TensorFlow 版本 2.7.0，请使用显示的通用方法 here:

import tensorflow as tf

# Convert the model
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir) # path to the SavedModel directory
tflite_model = converter.convert()

# Save the model.
with open('model.tflite', 'wb') as f:
  f.write(tflite_model)

返回结果并在遇到问题时标记我

Answer 2

你可以看看这个github(https://github.com/hunglc007/tensorflow-yolov4-tflite)，我想你会找到有用的东西