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