jhjin / tensorflow-cpp

Image classification C++ example using tensorflow shared library

Tensorflow C++ API example

The repository provides a basic image classification example using Tensorflow shared library (.so). Tested on the Ubuntu 16.04 machine.

Dependencies

Download cudnn library under the lib directory for CUDA.

lib/cudnn.h
lib/libcudnn.so.5  # with major version included in filename

and make dependency to install dependent packages via apt-get. Update CMakeLists.txt according to your configuration if needed.

Build and run

Calling the Makefile target will build tensorflow library, download a pretrained model, and run the app.

make

If you need python interface, try pip install lib/tensorflow*.whl.