ONNX-Parser is a C++ Inference Code Generator tool that takes an onnx binary model and generates OpenVX GDF code for quick prototyping and kernel debugging.
The details of various OpenVX Kernels generated are from vx_nn
Note: Currently only the float-32 tensor datatypes are supported. Special layers such as ROI Pooling Layer, Deconvolution are not yet supported.
Network Name | Supported |
---|---|
AlexNet | Yes |
VGG-16 | Yes |
VGG-19 | |
ResNet-50 |
% onnx_gdf_generator <net.pb> [n c H W]
Here net.pb is the onnx binary models which is a mandatory file. n,c,H,W refers to batch size, number of channels, height and width of an image which are optional parameters.
ONNX Binary models can be found here : models
- After cloning the repository, create a build folder where the executables has to be present.
- cmake -DCMAKE_BUILD_TYPE=Release ../onnx-parser
- make
Now, the executables are built and present in the build folder.