请我需要一个由yolov5生成的模型'last.pt'的示例，并用Java实现以检测对象。我想将它应用到我自己的模型上来构建一个 android APK。谢谢
问问题
754 次
1 回答
1
您需要的是分别支持移动推理的 PyTorch 或 Tensorflow 的移动优化版本、PyTorch Mobile 和 TensorflowLite。这个存储库就是这样做的。它运行 TFLite 并使用 yolov5 模型。yolov5 是家族中第一个在 PyTorch 中实现的，这一事实使这成为可能。
在模型文件夹中，您将找到一个脚本，它将您的 .pt 模型转换为 .tflite，TFLite 需要它作为权重格式。您将找到有关如何进行该转换的更多详细信息。该应用程序具有单页并执行实时推理，但当然您可以根据需要对其进行修改。
此拉取请求中有很多问题，如果您有其他问题，作者会积极回复。
于 2021-07-13T06:18:46.793 回答