yolo v5 导出 Android 手机上可以运行的 TFLite 模型

原文英文,约400词,阅读约需2分钟。发表于:

TFLite 是 TensorFlow Lite 的缩写。TFLite 在 Android 上的运行速度更快,因为它专门针对移动设备进行了优化。但是,最后我发现我实在找不到在 Android 上执行 yolov5 TFLite 模型的示例,转而使用 pytorch mobile 的 torchscr...

TFLite是TensorFlow Lite的缩写,用于优化移动设备。导出命令为python export.py --weights yolov5s.pt --include tflite,导出后缀为yolov5s.tflite。导出的tflite位于原模型同级目录。验证可用性的命令为!python detect.py --weights /content/drive/MyDrive/yolo/wheat_head_best-fp16.tflite --source /content/drive/MyDrive/yolo/wheat_heads。导出的tflite模型可用性测试结果显示检测到了多个人。

yolo v5 导出 Android 手机上可以运行的 TFLite 模型
相关推荐 去reddit讨论