search

cmdbug / TNN_Demo

🍉 移动端TNN部署学习笔记，支持Android与iOS。
GNU General Public License v3.0
71 stars 18 forks source link

yolov5s v5.0模型转换TNN后运行出错 #5

Open winterfell2021 opened 3 years ago

winterfell2021 commented 3 years ago

使用的模型：https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt 转换onnx命令：python models/export.py --weights yolov5s.pt --img 640 --batch 1 转换TNN命令：python converter.py onnx2tnn yolov5s.onnx -o . -in 1,3,448,640 转换后的模型：https://www.lanzoui.com/iSktTqkfote 运行demo时报错： image

cmdbug commented 3 years ago

有v3.0版本的转换过程，v5版本的没试过。 https://github.com/cmdbug/TNN_Demo/issues/1