YOLOv5_Android_USBCamera:YOLOv5模型定制化及转换

本文是之前项目的说明,用以替换定制化模型
前文:YOLOv5_Android_USBCamera:支持USB摄像头的YOLOv5Android图像识别项目


模型转换过程

根据原项目的描述 https://github.com/lp6m/yolov5s_android ,这个项目使用的模型是通过从PyTorch原始实现的模型转换获得一个TFlite模型,而不是在TFlite中自己实现。

官方的模型定制化说明(必读):https://github.com/lp6m/yolov5s_android/issues/14

总体的转换过程为PyTorch -> ONNX -> OpenVino -> TFLite

步骤

  • PyTorch (.pt 文件)
    • 用实验室已有的YOLO项目源码可以得到模型权重文件
  • ONNX (.onnx文件)
    • 简介:https://blog.csdn.net/gaoxueyi551/article/details/117331600
    • 转换:https://blog.csdn.net/weixin_38989668/article/details/123840882
  • OpenVino
  • TFLite
    • 将 OpenVino 模型转换为 TFLite,作者使用的工具:
      https://github.com/PINTO0309/openvino2tensorflow
    • 参考文档:
      https://github.com/lp6m/yolov5s_android/tree/master/convert_model
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值