wangshankun / multithreading_frcnn

multithreading faster rcnn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

faster-rcnn: multithreading for caffe:pooling_layer,relu_layer,roi_pooling_layer,im2col

8 threads CPU(i7-6700)layer speed up x5 ~ x7,(im2col(x2.5) DDR4 is the bottlenecks)

Add fucntion

  • add multithreading for caffe:

    modified: src/caffe/layers/pooling_layer.cpp

    modified: src/caffe/layers/relu_layer.cpp

    modified: src/caffe/layers/roi_pooling_layer.cpp

    modified: src/caffe/net.cpp

    modified: src/caffe/util/im2col.cpp

    add: include/caffe/calcu_pthread.h

    add: src/caffe/calcu_pthread.cpp

    add: Makefile.config

  • add westwell port project using modle:

    modified: tools/demo.py

    add: westwell/

Compile:

  • cd multithreading_frcnn/caffe-fast-rcnn

    make -j8 && make pycaffe

  • cd multithreading_frcnn/lib

    make

Exec:

python ./tools/demo.py --cpu

About

multithreading faster rcnn

License:Other


Languages

Language:Jupyter Notebook 47.9%Language:C++ 38.6%Language:Python 7.8%Language:Cuda 2.8%Language:CMake 1.4%Language:MATLAB 0.5%Language:Shell 0.4%Language:Makefile 0.3%Language:C 0.1%Language:CSS 0.1%Language:HTML 0.1%