如何在 iOS 应用程序中使用自定义 yolov4 权重

发布于 2025-01-10 11:49:22 字数 695 浏览 1 评论 0原文

我按照以下步骤操作:

  1. 在自定义数据集上训练后获得 YOLO v4 权重
  2. 将权重转换为 TensorFlow 检查点
  3. 将 Tensorflow 检查点转换为 tflite
  4. 量化 tflite
  5. 我使用的是 tflite 示例 和将自定义 tflite 导入到 xcode 项目

但我遇到了崩溃:

Failed to create the interpreter with error: Failed to create the interpreter.
ObjectDetection/ViewController.swift:56: Fatal error: Failed to load model
2022-02-28 13:28:22.850305+0530 ObjectDetection[9505:3901451] 
ObjectDetection/ViewController.swift:56: Fatal error: Failed to load model

我测试了我的 tflite 权重及其工作正常

I following these steps:

  1. Obtained YOLO v4 weights after training on my custom dataset
  2. Converted weights to TensorFlow checkpoints
  3. Converted Tensorflow checkpoint to tflite
  4. Quantized tflite
  5. I was using tflite samples and imported custom tflite to xcode project

But I'm getting a crash:

Failed to create the interpreter with error: Failed to create the interpreter.
ObjectDetection/ViewController.swift:56: Fatal error: Failed to load model
2022-02-28 13:28:22.850305+0530 ObjectDetection[9505:3901451] 
ObjectDetection/ViewController.swift:56: Fatal error: Failed to load model

I tested my tflite weights and its working fine

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文