将onnx型号转换为apple coreml格式。
这里有一个全面的教程
安装
从pypi安装pip install -U onnx-coreml
从源安装
要获取转换器的最新版本,请从源代码安装,方法是克隆存储库及其子模块并运行install.sh脚本。也就是说,git clone --recursive https://github.com/onnx/onnx-coreml.git
cd onnx-coreml
./install.sh
从源安装(供贡献者使用)
要获取转换器的最新版本,请从源代码安装,方法是克隆存储库及其子模块并运行install-develope.sh脚本。也就是说,git clone --recursive https://github.com/onnx/onnx-coreml.git
cd onnx-coreml
./install-develop.sh
依赖关系单击
努比
CoremlTools(2.0+)
onnx(1.3.0+)
如何使用
要转换模型,请使用onnx coreml中的单个函数"convert":fromonnx_coremlimportconvertdefconvert(model,mode=None,image_input_names=[],preprocessing_args={},image_output_names=[],deprocessing_args={},class_labels=None,predicted_feature_name='classLabel',add_custom_layers=False,cu