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格式。

你可能感兴趣的:(pytorch,深度学习)