1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
Caffe2 implementation of Open Neural Network Exchange (ONNX)
========
# Usage
* [ONNX to Caffe2](https://github.com/onnx/tutorials/blob/master/tutorials/OnnxCaffe2Import.ipynb)
* [Caffe2 to ONNX](https://github.com/onnx/tutorials/blob/master/tutorials/Caffe2OnnxExport.ipynb)
* [other end-to-end tutorials](https://github.com/onnx/tutorials)
# Installation
onnx-caffe2 is installed as a part of Caffe2.
Please follow the [instructions](https://caffe2.ai/docs/getting-started.html) to install Caffe2.
# Folder Structure
- ./: 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
# Testing
onnx-caffe2 uses [pytest](https://docs.pytest.org) 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.
Testing coverage issues/status: https://github.com/caffe2/caffe2/blob/master/caffe2/python/onnx/ONNXOpCoverage.md
# Development
During development it's convenient to install caffe2 in development mode:
```
cd /path/to/caffe2
pip install -e caffe2/
```
# License
[MIT License](LICENSE)
|
