zacario-li / caffe2_Cpp_example

c++ caffe2 example

caffe2_Cpp_example

c++ caffe2 example

Usage:

** notice **

before you compile this code, please double check your pytorch version, I use pytorch master tag v0.4.1, do not use the latest commit
cd 'project root dir'
mkdir build
cd build
cmake ..
make
'run and have fun'

---

Project description

use caffe2 c++ api to load pretrained model.
I use a mp4 file to test squeeznet with CUDA.
Next step, I will try to modify this code running with opencl.

Notice:

Build caffe2 with opencv which already installed on your system(mine is ubuntu 16.04)
If you use CLion to build the project, maybe you will meet DSO missing from command line ERROR, double check clion's link.txt file, maybe the command line is missing some libraries.