使用 YOLO4 模型在 Android 上运行对象检测是不是可行？

Posted 2023-03-16

【中文标题】使用 YOLO4 模型在 Android 上运行对象检测是不是可行？

【英文标题】：Is running object detection on Android using YOLO4 model possible at all?使用 YOLO4 模型在 Android 上运行对象检测是否可行？

【发布时间】：2021-11-14 23:10:12

【问题描述】：

我已经使用 YOLOv4 训练了自定义对象检测模型。使用 Python 成功运行推理后，我将在 android 设备上进行推理。所以我将它转换为 TFLite 16bit 量化格式。但是当我使用 Tensorflow lite 任务视觉库加载它时，我不断收到错误：

'Mobile SSD models are expected to have exactly 4 outputs, found 3'.

据我了解，此错误是由于模型与库的要求不兼容而导致的。但是由于输出的数量是特定于 YOLOv4 模型架构的，这是否可行？

任何帮助将不胜感激。

最好的问候， 设置Nug

【问题讨论】：

从任务库tensorflow.org/lite/inference_with_metadata/task_library/…检查这些要求如果它们不适合你，你必须使用普通的解释器在设备上进行推理。

似乎这是唯一的出路。谢谢

【参考方案1】：

是的，但推理速度极慢

【讨论】：

您的答案可以通过额外的支持信息得到改进。请edit 添加更多详细信息，例如引用或文档，以便其他人可以确认您的答案是正确的。你可以找到更多关于如何写好答案的信息in the help center。