Skip to content

black-og/segmentation-caffe-cpp

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Do inference for semantic segmentation with trained caffe models.

  1. Put the caffemodel and deploy file in the models folder. Then the deploy file is named deploy.prototxt, and the caffemodel file is named segment.caffemodel. Alternatively, you can also change the relevant parameters in the src/caffe_cpp_demo.cpp to the corresponding path:

    string modelFile = the path of your deploy file;

    string trainedFile = the path of your caffemodel file;

  2. Put the images in the data folder. Or, you can change the parameter in the src/caffe_cpp_demo.cpp to the corresponding path:

    string image_path = the path of your images;

  3. Change INCLUDE_DIRECTORIES and LINK_DIRECTORIES for cuda, opencv, caffe, boost, gflags, and glog in CMakeLists.txt.

  4. Change working directory to current directory.

  5. cd build

  6. cmake ..

  7. make

  8. sudo ./segDemoCPP

About

Segmentation with trained caffe models, it is written in C++.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published