Skip to main content

Convert Caffe models to ONNX.

Project description

caffe2onnx

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

Installation

Install from pypi

pip install caffe2onnx

Install latest from github

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

Build and install latest from source

git clone https://github.com/asiryan/caffe2onnx
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).
python -m caffe2onnx.convert
    --prototxt          caffe prototxt file path
    [--caffemodel       caffe caffemodel file path]
    [--onnx             output onnx file path]
    [--frozen           frozen graph or not]

Operators

See the documentation of caffe supported operators.

References

caffe-onnx by htshinichi
TNN by Tencent

License

BSD-3

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

caffe2onnx-2.0.1.tar.gz (49.9 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page