1、下载pnnx

    去github下载预编译好的版本

2、导出模型

# 1. 导出TorchScript
yolo export model=best.pt format=torchscript imgsz=320

# 2. 转换
./pnnx best.torchscript inputshape=[1,3,320,320]

测试结果很奇怪:原来opencv dnn跑出来的平均9ms,ncnn跑出来12ms,反而变慢了!

yolo export model=best.pt format=torchscript imgsz=96 simplify=True half=False device=cpu
/pnnx best.torchscript inputshape=[1,3,96,96] fp16=0 optlevel=2

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐