Skip to content

Latest commit

 

History

7 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

Image classification C++ example using tensorflow shared library

Topics

Resources

Stars

103 stars

Watchers

10 watching

Forks

Releases

Packages

Contributors

Languages