Skip to content

faster rcnn cuda 10

2018.10.06 00:09

WHRIA 조회 수:2101

https://gist.github.com/acmiyaguchi/bc535ba23eabd3564edd73e491763d50

 


이 사람의 rebased 를 download 하고 나서 아래처럼 patch

 

Instructions

  • install cuda-9.0 (https://developer.nvidia.com/cuda-toolkit-archive)
  • install cudnn-7 (https://developer.nvidia.com/cudnn)
  • install dependencies for caffe
    • see the caffe installation guide [1] and wiki page [2]
  • install boost-1.65 or higher (http://www.boost.org/users/history/version_1_65_1.html) [3]
  • update caffe-fast-rcnn
    • Easiest way is to use a rebased branch on BVLC/master. See the notes below if you would rather merge the changes yourself.
    $ cd $FRCN_ROOT/caffe-fast-rcnn
    $ git remote add acmiyaguchi https://github.com/acmiyaguchi/caffe-fast-rcnn.git
    $ git checkout acmiyaguchi/faster-rcnn-rebased
    
    • optional: rebase against upstream master
    $ git remote add caffe https://github.com/BVLC/caffe.git
    $ git rebase caffe/master
    
  • set variables in Makefile.config
    • USE_CUDNN := 1
    • WITH_PYTHON_LAYER := 1
    • OPENCV_VERSION := 3
    • CUSTOM_CXX := g++-6
      • I symlink gcc-6 and g++-6 to gcc and g++ in /usr/local/cuda-9-0/bin/ instead of this setting, but either should work
    • remove compute_20 from CUDA_ARCH since it's been depreciated in cuda-9-0
    • add hdf5 for linking since it's been renamed in 17.10
      • INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/include/hdf5/serial/
      • LIBRARY_DIRS := $(PYTHON_LIB) /usr/lib/x86_64-linux-gnu/hdf5/serial/
  • build caffe
    • make -j${N_THREADS} all && make pycaffe
  • change self.param_str_ to self.param_str in $FRCN_ROOT/lib/rpn/proposal_layer.py.
    • see rebased-caffe.patch below for the full diff
  • validate that ./tools/demo.py works

Notes on Rebasing

BVLC/caffe and rbgirshick/caffe have diverged since this workaround [4] in March 2017. I've rebased the faster-rcnnpatchset against the master caffe branch for convenience. [5]

This branch should merge cleanly into BVLC/caffe with the exception of the license.

If you rather merge the upstream changes yourself, the patch in [6] should outline the changes you need afterwards. -X theirs may not be the best merge policy. In particular, changes to the src/caffe/caffe.proto schema will overwrite the ROIPoolingParameter.

rebased-caffe.patch

diff --git a/lib/rpn/proposal_layer.py b/lib/rpn/proposal_layer.py
index b157160..e6d28cc 100644
--- a/lib/rpn/proposal_layer.py
+++ b/lib/rpn/proposal_layer.py
@@ -23,7 +23,7 @@ class ProposalLayer(caffe.Layer):
 
     def setup(self, bottom, top):
         # parse the layer parameter string, which must be valid YAML
-        layer_params = yaml.load(self.param_str_)
+        layer_params = yaml.load(self.param_str)
 
         self._feat_stride = layer_params['feat_stride']
         anchor_scales = layer_params.get('scales', (8, 16, 32))
번호 제목 글쓴이 날짜 조회 수
274 usb 3.1 + DP WHRIA 2020.01.07 168
273 resnest50 caffe WHRIA 2020.06.21 166
272 concat network [3] WHRIA 2020.08.27 166
271 pytorch pretrained WHRIA 2020.10.28 166
270 CORS WHRIA 2021.01.22 166
269 페이닥터 등록 file WHRIA 2014.10.21 162
268 pROC package [3] WHRIA 2019.05.11 160
267 Transformer WHRIA 2020.10.09 160
266 fast rcnn cudnn 5.0 WHRIA 2017.01.08 157
265 yolo custom WHRIA 2019.07.25 157
264 pytorch optimize WHRIA 2020.08.01 156
263 역시 개원해서는 돈 버는 재미밖에 없다. WHRIA 2016.06.17 155
262 SAS controller SAS 9212-4i WHRIA 2019.12.08 155
261 stuff [2] WHRIA 2019.07.15 153
260 startup , 미국 WHRIA 2020.07.30 153

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


이 PC에는 나눔글꼴이 설치되어 있지 않습니다.

이 사이트를 나눔글꼴로 보기 위해서는
나눔글꼴을 설치해야 합니다.

설치 취소