Skip to content
This repository has been archived by the owner on Mar 1, 2022. It is now read-only.

asiryan/caffe2onnx

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

57 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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