yolov5转onnx到ncnn

测试代码6.2
检测这一套都没啥说的主要在onnx转ncnn这步

  1. python export.py --data data/xuehua.yaml --weights runs/train/exp4/weights/best.pt --train (这里是pt转onnx)
  2. python -m onnxsim runs/train/exp4/weights/best.onnx runs\train\exp4\weights\best-sim.onnx(这步重要),如果忘了用这个https://convertmodel.com/ 转ncnn估计也好使,理解是把onnx里面ncnn不支持的算子都去除掉
    3.下载ncnn的转换工具,有windows版本的 下载地址:https://github.com/Tencent/ncnn,选这个版本里面找onnx2ncnn的exe,通过cmd命令启动,转换命令:onnx2ncnn.exe xx.onnx
    这样就会得到一个转换好的一个bin和param结尾的文件
    4.用pycharm工具打开param结尾的文件,把里面的Reshape 后面对应的0=6400、0=1600、0=400均修改为0=-1(这步重要,不能少) 在这里-1表示自动计算,到这里就算是转换完成了
    网上找了一个做app程序的代码uniapp,搜索yolov5的插件,https://ext.dcloud.net.cn/plugin?id=5243
    这个老哥已经上传,我也是请教他的,把模型和类别修改为直接的打包成app就可以实现app里面使用yolov5做目标检测了
    onnx转ncnn错误记录
    在这里插入图片描述

这里就是提示有些算子不支持,其实在官方的文档里面也有说,只是如果不注意看也不知道给出的解决办法是去这个网址线上转换,其实最后就会跳转到https://convertmodel.com/这个网址,然后上传onnx模型一键转换成ncnn
在这里插入图片描述

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值