yolov5移动端部署.pt文件转.onnx文件

Question

背景：最近在做目标检测，其中用到的yolov5模型，框架是pytorch，由于想要把模型部署到移动端，因此需要模型格式的转换，思路如题目：pt > ONNX > CoreML > tflite
转换时遇到了问题，将原始模型文件.pt转化为onnx和torchscript都成功了

但转化为coreml时失败了，由于各种原因，这是最不能失败的一个，失败原因如下：
Starting ONNX export with onnx 1.11.0...
ONNX export success, saved as E:\yolov5\yolov5-v5.0\yolov5-v5.0/runs/train\exp34\weights/best.onnx
CoreML export failure: No module named 'coremltools.libmilstoragepython'

我在pip list 时发现我是有coremltools 6.1 的版本的，想请问一下各位 这是什么原因？

Answer 1

我是没看懂为什么你要的tflite和转出CoreML有啥关系，源码可以直接转到tflite不需要经过CoreML ，export_tflite（）这个函数里面就没有调用到CoreML的相关包