MIVisionX DGtest is a tutorial program for those who are new to MIOpen & OpenVX. It runs inference on the MNIST dataset using MIOpen & OpenVX.
The program is divided into four parts:
annmodule is created by the AMD model compiler. You can convert your own trained caffemodel to an openvx graph using the model-compiler. See the below section for the detailed instruction. Once the conversion is done, the corresponding c++ codes will be generated including annmodule which contains the information about your pre-trained caffemodel.
The user-interface class is for the GUI functionality of the DGtest. It has the palette and the result window to draw the image and show the resulting output.
DGtest class is where you run the inference. Using the annAddToGraph() function in annmodule, it adds weights, input tensor, and output tensor to the graph. If it was successful, it preprocesses the image and processes the graph (runs the inference) using the vxProcessGraph() function.
See the OpenVX documentation for detailed explanation about OpenVX API calls.
- Build & Install MIVisionX
- OpenCV 3.1 or higher
Build using Cmake on Linux (Ubuntu 16.04 64bit)
mkdir build cd build cmake .. make
weights.bin: The name of the weights file to be used for the inference. It is created by running a caffemodel converter. See the below section for using your caffemodel.
Testing with your Caffemodel
You can test your trained MNIST caffemodel using the model compiler
- Convert your caffemodel->NNIR->openvx using the model compiler.
- From the generated files, copy
annmodule.cpp annmodule.h weights.bin
to the DGtest folder.
- Build the program again.