在java中集成yolov5 for android APK

Posted 2023-03-12

【中文标题】在java中集成yolov5 for android APK

【英文标题】：integring yolov5 in java for android APK

【发布时间】：2021-09-08 18:49:39

【问题描述】：

我需要一个由 yolov5 生成的模型“last.pt”示例，并用 Java 实现以检测对象。 我想将它应用到我自己的模型上来构建一个 android APK。 谢谢

【参考方案1】：

您需要的是支持移动推理的 PyTorch 或 Tensorflow 的移动优化版本，分别是 PyTorch Mobile 和 TensorflowLite。 This repository 就是这样做的。它运行 TFLite 并使用 yolov5 模型。 yolov5 是家族中第一个最初在 PyTorch 中实现的事实，这使得这成为可能。

在模型文件夹中，您将找到一个脚本，该脚本将您的 .pt 模型转换为 TFLite 需要的 .tflite 作为权重格式。您将找到有关如何进行该转换的更多详细信息。该应用程序具有单页并执行实时推理，但您当然可以随意修改它。

这个pull request有很多问题，如果您还有其他问题，作者会积极回复。