yolov3android,android - TensorFlow Lite - Object Detection API YOLOv3 - Stack Overflow

博客讨论了将完整的YOLOv3模型转换为TensorFlow Lite模型的问题,由于YOLOv3头部部分的额外层,目前无法直接转换。建议可以转换不包含头部的YOLOv3模型,但需要在Java代码中实现缺失部分。同时,推荐使用SSD-MobileNet或YOLOv2-tiny作为实时应用的替代方案。还提到了从1.14版之后,需要更新导入库的方式以适配TensorFlow Lite。
摘要由CSDN通过智能技术生成

Unfortunately you can't convert the complete YOLOv3 model to a tensorflow lite model at the moment. This is because YOLOv3 extends on the original darknet backend used by YOLO and YOLOv2 by introducing some extra layers (also referred to as YOLOv3 head portion), which doesn't seem to be handled correctly (atleast in keras) in preparing the model for tflite conversion.

You can convert YOLOv3 to .tflite without the model's 'head' portion (See here: https://github.com/benjamintanweihao/YOLOv3), but then you will have to implement the missing parts in your Java code (as suggested here: https://github.com/wics1224/yolov3-android-tflite). Make sure you have the correct anchor box sizes if you do so. The second link would hopefully answer the second part of your question.

If you plan to keep things simple, your other options would be using SSD-mobilenet or yolov2-tiny for your application. They will give you a more real-time experience.

I am currently working on a similar project involving object detection in flutter/tflite so I'll keep you updated if I find anything new.

Edit:

In https://github.com/benjamintanweihao/YOLOv3, you'll need to change how you import libraries because lite library is moved out from contrib from tensorflow 1.14 onwards.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值