caffe2onnx 工具使用说明

1. caffe2onnx 工具使用说明#

本章介绍 AX 版 caffe2onnx 转换工具, 用于将 Caffe 浮点模型转换成 ONNX 浮点模型. 因为平台的限制, 目前 X86 平台跟 ARM 平台 发版提供的命令行接口有一些区别.

备注

下文的模型语义皆为浮点模型.

1.1. 将 Caffe 模型转换成 ONNX 模型#

我们提供了命令行工具将 Caffe 模型转换成 ONNX 模型. 在 X86 平台下, 您可以传入一个文件夹以转换其里面所有的 Caffe 模型：

caffe2onnx_cli --convert --checkpoint_path /path/to/your/model/zoo

这将递归地找到指定文件夹里面所有的以 ".caffemodel" 为后缀文件及其对应的 ".prototxt" 文 件, 此为一个 Caffe 模型, 将其转换为 ONNX 模型, 并使用 Caffe 模型的 前缀, 以 ".onnx" 为后缀进行保存.

备注

Caffe 模型对应的 ".prototxt" 和 ".caffemodel" 文件需要在同一个文件夹并共享一个前缀.

1.2. 验证转换出的 ONNX 模型#

在 X86 平台下, 您可以使用如下命令行工具对分原始的 Caffe 模型和转换出的 ONNX 模型：

caffe2onnx --validate --checkpoint_path /path/to/your/model/zoo

首先这将递归地找到指定文件夹里所有的以 ".onnx" 为后缀的文件, 然后按照其前缀匹配对应的 ".prototxt" 和 ".caffemodel" 文件, 生成一个随机数据集, 分别使用 ONNXRuntime 和 Caffe 推理工具进行推理, 并计算两者的 “相关系数 (Correlation)”、“标准偏差 (Standard Deviation)”、 “余弦距离相似度 (Cosine Similarity)”、“归一化相对误差 (Normalized Relative Error)”、 “最大差异 (Max Difference)” 和 “平均差异 (Mean Difference)”.

备注

Caffe 模型对应的 ".prototxt" 和 ".caffemodel" 文件以及转换出来的 ".onnx" 文件需要在同一个文件夹并共享一个前缀.

备注

此步限制标准 Caffe 算子.

备注

因为 Caffe ARM 平台的兼容性不太好, ARM 平台下暂时不支持这个功能.

目录