当前位置:   article > 正文

yolov3(pytorch)转 ncnn 详解 |【❤️yolov3-spp.pt 转 onnx 转 ncnn❤️】_yolov3 ncnn

yolov3 ncnn

yolov3-spp.pt 【 pytorch 】模型 转 onnx 转 ncnn 推理


yolov3-spp.pt ==>> yolov3-spp.onnx ==>> yolov3_spp_pt.param 【这一步卡壳,暂无方案】


思路一【使用 yolov5 下的 yolov5-5.0/models/export.py 】转 onnx


yolov5 模型转换之 PyTorch to ONNX 博文

# 下载 yolov5-5.0 版本代码

cd yolov5-5.0

# 把 yolov3 模型拷贝到 该目录下

ll yolov3weights/

total 492784

drwxrwxr-x  2 zql zql      4096 Sep 24 07:06 ./
drwxrwxr-x 10 zql zql      4096 Sep 24 07:05 ../
-rw-rw-r--  1 zql zql 252297587 Sep 24 07:05 yolov3-spp.pt
-rw-rw-r--  1 zql zql 252297867 Sep 24 07:05 yolov3-spp-ultralytics.pt

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15

遇到 AttributeError 报错如下 【暂不处理】

# 运行命令

python models/export.py --weights ./yolov3weights/yolov3-spp.pt --img 608 --batch 1

# 输出如下

Namespace(batch_size=1, device='cpu', dynamic=False, grid=False, img_size=[608, 608], weights='./yolov3weights/yolov3-spp.pt')
YOLOv5 
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/558527
推荐阅读
相关标签