(三)Yolov8的NCNN模型导出与关键点检测任务部署

(三)Yolov8的NCNN模型导出与部署


欢迎访问个人网络日志🌹🌹知行空间🌹🌹


1.yolov8导出ncnn模型

1).ultralytics直接导出ncnn模型

from ultralytics import YOLO

# Create a model
model = YOLO('/xx/data/code/ultralytics/yolov8n-pose.pt')

# Export the model to NCNN with arguments
model.export(format='ncnn', half=True, imgsz=480)

执行上面的代码,会在当前路径创建yolov8n-pose_ncnn_model文件夹:

.
├── metadata.yaml
├── model.ncnn.bin
├── model.ncnn.param
└── model_ncnn.py

model_ncnn.pyncnn模型推理的测试文件,可以直接执行测试导出的模型是否能正常工作。

不过ultralytics是使用pnnx导出的ncnn模型,使用c++导入,

auto net_ = std::make_unique<ncnn::Net>();
assert(net_->load_param(param_file) == 0);
assert(net_->load_model(bin_file) == 0);

2).ultralytics导出onnx模型再转ncnn模型

step1:ultralytics导出onnx模型

from ultralytics import YOLO

# Create a model
model = YOLO('/xx/data/code/ultralytics/yolov8n-pose.pt')

# Export the model to NCNN with arguments
model.export(format='ncnn', half=True, imgsz=
  • 21
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值