Skip to content

Image classification C++ example using tensorflow shared library

Notifications You must be signed in to change notification settings

jhjin/tensorflow-cpp

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

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

Watchers

Forks

Releases

No releases published

Packages

No packages published