Skip to content

faster rcnn cuda 10

2018.10.06 00:09

WHRIA 조회 수:2109

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))
번호 제목 글쓴이 날짜 조회 수
1114 잘 해주는 사람 vs 잘 해주고 싶은 사람 [1] WHRIA 2006.04.24 2080
1113 회식 han 2006.05.20 2080
1112 불면증 han 2006.05.30 2080
1111 Re: 저승사자가 된것 같아요. 한승석 2003.02.07 2081
1110 동일시 한승석 2003.02.07 2081
1109 헤헤.. 형두 들어오시는군요. 한승석 2003.03.04 2081
1108 따분하다. Whria 2003.05.06 2081
1107 휴식 Ideon 2003.05.12 2081
1106 일처리 한승석 2003.05.30 2081
1105 발표 그리고 발표 그리고 또 발표 NEWTYPE 2003.06.06 2081
1104 LOVE NEWTYPE 2003.06.06 2081
1103 일년차들 빨리 들어왔으면 좋겠다. 한승석 2004.01.10 2081
1102 아토피 피부염 환자 NEWTYPE 2004.01.28 2081
1101 2 가지 업데이트 한승석 2004.02.28 2081
1100 오늘은 내가 보는 외래가 있는날 Han 2004.04.27 2081

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소