Skip to content

Caffe2 implementation of Open Neural Network Exchange (ONNX)

License

Notifications You must be signed in to change notification settings

takeontom/onnx-caffe2

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

onnx-caffe2

Caffe2 implementation of Open Neural Network Exchange (ONNX).

Repository location may change.

Folder Structure

  • onnx_caffe2/: the main folder that all code lies under
    • frontend.py: translate from caffe2 model to onnx model
    • backend.py: execution engine that runs onnx on caffe2
  • tests/: test files

Installation

pip install onnx-caffe2

Testing

onnx-caffe2 uses pytest as test driver. In order to run tests, first you need to install pytest:

pip install pytest-cov

After installing pytest, do

pytest

to run tests.

Development

During development it's convenient to install onnx-caffe2 in development mode:

git clone  https://github.com/onnx/onnx-caffe2.git
pip install -e onnx-caffe2/

License

MIT License

About

Caffe2 implementation of Open Neural Network Exchange (ONNX)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 95.6%
  • Shell 4.4%