- ONNX:该模型格式可以在不同的平台所需的模型格式之间进行转化,也是我最初的方案,但转过去转过来都转不到Android能用的格式
- CoreML:该格式的模型可以部署到Apple上,那至少需要有苹果两件套(iphone和Mac),前者作为部署端,后者用于开发苹果应用程序。可是我没有Mac
- TFLite:Google的轻量级推理库,这种TensorFLow Lite格式的模型文件可以部署到基于Android的移动端上,这也是最理想的一种方式
转化方法
官方的转化方法
官方的转化代码的转化方式为PyTorch > ONNX > CoreML > TFLite,最后从原始训练得到的pt模型文件转化为mlmodel、onnx和torchscript.pt
不过要想从这三种模型进行二次转化为**.tflite**可不容易,博主尝试过很多种方法,但最终无果,也不是走投无路,就是转累了。。。