This repo contains a header-only library to load and test a pre-trained neural network.
| File/Folder name | Usage |
|---|---|
| cmake/ | Contains cmake helper files |
| data/ | Contains data file (e.g. Pre-trained network, dataset) |
| include/ | Contains all header files |
| src/ | Contains file for pybind |
| test/ | Contains unit tests |
| CMakeLists.txt | CMakeLists file |
| syntax.txt | The syntax of pre-trained network definition file |
This library can load pre-trained neural network, load dataset, and run test on dataset (tested on MNIST dataset).
Eigen, Pybind11
Under project root directory
git clone https://github.com/pybind/pybind11.git
mkdir build
cd build
cmake ..
make
Executable should be generated in build folder.