ONNX和NCNN都是什么?

ONNX

-ONNX是一种模型存储方式,类似于编程中使用的JSON数据格式,与语言和平台无关

  • 很多机器学习平台都支持把自身训练的模型转为ONNX格式的模型
  • ONNX由微软、亚马逊、Facebook、IBM共同开发
  • 后缀是*.proto

NCNN

  • 是腾讯开发的一个能够把模型部署到移动设备上的部署框架
  • 如果要ONNX模型部署到Android设备上,也需要NCNN来进行部署,而不是直接部署ONNX

总结

所以如果想要把detectron2的模型部署到Android设备上,需要先把.pth模型转换为ONNX模型,然后再通过NCNN把ONNX部署到Android设备上。

.pth > ONNX > ncnn

其中.pth转ONNX是detectron2自带的方法,而ONNX要转换成ncnn识别的文件格式就需要用到ncnn自带的小工具来转换,转成ncnn支持的.bin和param格式。


版权声明:本文为zhangdongren原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。