yolo11模型文件pt转ncnn可以简单的用直接转换,但已用的为参考网上大佬
https://github.com/nihui/ncnn-android-yolo11
转换的bin文件在android端部署。参数与直接转换的有所不同。
其步骤为
pip install -U ultralytics pnnx ncnn #安装
yolo export model=yolo11n.pt format=torchscript #导出
pnnx yolo11n.torchscript #转换
为适应不同大小的图片,yolo11n_pnnx.py 文件要修改
v_235 = v_204.view(1, 144, 6400)
v_236 = v_219.view(1, 144, 1600)
v_237 = v_234.view(1, 144, 400)
v_238 = torch.cat((v_235, v_236, v_237), dim=2)
改为
v_235 = v_204.view(1, 144, -1).transpose(1, 2)
v_236 = v_219.view(1, 144, -1).transpose(1, 2)
v_237 = v_234.view(1, 144, -1).transpose(1, 2)
v_238 = torch.cat((v_235, v_236, v_237), dim=1)

最低0.47元/天 解锁文章
1805






