项目作者: asiryan

项目描述 :
Convert Caffe models to ONNX.
高级语言: Python
项目地址: git://github.com/asiryan/caffe2onnx.git
创建时间: 2021-05-10T15:12:39Z
项目社区:https://github.com/asiryan/caffe2onnx

开源协议:BSD 3-Clause "New" or "Revised" License

下载


caffe2onnx

This tool converts Caffe models to ONNX via command line (without Caffe environment).

Installation

Install from pypi

  1. pip install caffe2onnx

Install latest from github

  1. pip install git+https://github.com/asiryan/caffe2onnx

Build and install latest from source

  1. git clone https://github.com/asiryan/caffe2onnx
  2. python setup.py install

Usage

To get started with caffe2onnx, run the caffe2onnx.convert command, providing:

  • the path to your caffe prototxt,
  • the path to your caffe model (not required),
  • the output path of the onnx model (not required),
  • frozen graph or not (not required).
  1. python -m caffe2onnx.convert
  2. --prototxt caffe prototxt file path
  3. [--caffemodel caffe caffemodel file path]
  4. [--onnx output onnx file path]
  5. [--frozen frozen graph or not]

Operators

See the documentation of caffe supported operators.

References

caffe-onnx by htshinichi
TNN by Tencent

License

BSD-3