yolov11模型在Android设备上运行【踩坑记录】

gwd777

已于 2025-02-07 02:17:57 修改

阅读量698

点赞数 1

文章标签： YOLO android python

于 2025-02-07 01:51:59 首次发布

本文链接：https://blog.csdn.net/gwd777/article/details/145484633

版权

0) 参考资料:

https://github.com/Tencent/ncnn?tab=readme-ov-file 
https://github.com/pnnx/pnnx
https://github.com/nihui/ncnn-android-yolov5
https://github.com/Tencent/ncnn?tab=readme-ov-file

1) ：将xxx.pt模型转化成 xxx.onnx

ONNX（Open Neural Network Exchange）是一种开放格式，可以让你在不同深度学习框架之间转换和共享模型 ; ONNX 适用于边缘设备和移动端,部署到 移动设备（如 Android / iOS）; Android 不能直接使用 ONNX 模型，但可以通过 ONNX Runtime 或 转换为 NCNN / TFLite /

在Linux环境下执行:
    pip install ultralytics 'onnx>=1.12.0', 'onnxslim', 'onnxruntime'
    python convert_pt2onnx() 得到 xxx.onnx 模型

def convert_pt2onnx():
    OreModelPath = 'E:\code_py_workspace\mainApp\\v8n-pt_oreDusty320Cls1_imgsz640.pt'
    model = YOLO(OreModelPath)

    model.eval()

    # first way of convertion
    path = model.export(format="onnx