Skip to content

Self Maintained Caffe. In this version Faster-RCNN, RFCN needs layer are fully supported!

Notifications You must be signed in to change notification settings

UnrealVision/unreal_caffe

Repository files navigation

Unreal Caffe

Travis Build Status Appveyor Build status Coverage Status Coverity Scan Build Documentation Status Gitter

unreal_caffe is a self-maintained caffe version without pain in official repo. current version support some layer which needed by Faster-RCNN and R-FCN in C++. CUDNN version update to cudnn7.

  • cudnn 7 support, add to cudnn7;
  • roi_pooling_layer needed in Fast-RCNN series task;
  • smooth_l1_loss_ohem_layer needed in RFCN task;
  • deformalable_conv_layer which is the most advanced conv (still under test.);

Note: the default config in Makefile.config not using CUDA and CUDNN, if you wanna build on CUDA, make sure you have CUDA9.0 & CUDNN 7 installed.

Install

For installation, pls install depencies first incase you are not already for the installation:

./install_depencies.sh
make all -j32
sudo make install

Contribute

If you wanna contribute to this project, you can add my wechat: jintianiloveu. New layers and feature and examples are welcome!

Using in your project

Do something like this in your project.

link_directories("~/unreal_caffe/include/")
include_directories("~/unreal_caffe/include/")
find_library(caffe_LIBRARY
        NAMES  caffe
        HINTS "~/unreal_caffe/build/lib/"
        PATHS "~/unreal_caffe/build/lib/")

target_link_libraries(rfcn_caffe ${caffe_LIBRARY})

Copyright

Original copyright:

@article{dai16rfcn,
    Author = {Jifeng Dai, Yi Li, Kaiming He, Jian Sun},
    Title = {{R-FCN}: Object Detection via Region-based Fully Convolutional Networks},
    Journal = {arXiv preprint arXiv:1605.06409},
    Year = {2016}
}

About

Self Maintained Caffe. In this version Faster-RCNN, RFCN needs layer are fully supported!

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published