Skip to content

fast rcnn cudnn 5.0

2017.01.08 15:47

WHRIA 조회 수:128

手动修改文件,参考了卜居大神的博客 http://blog.csdn.net/kkk584520/article/details/51163564

方案1简单方便,但是当我们编译的是其他人修改过得源码,可能就会出错。方案2 步骤如下:

 

1. 用最新caffe源码的以下文件替换掉faster rcnn 的对应文件

 

include/caffe/layers/cudnn_relu_layer.hpp, src/caffe/layers/cudnn_relu_layer.cpp, src/caffe/layers/cudnn_relu_layer.cu

 

include/caffe/layers/cudnn_sigmoid_layer.hpp, src/caffe/layers/cudnn_sigmoid_layer.cpp, src/caffe/layers/cudnn_sigmoid_layer.cu

 

include/caffe/layers/cudnn_tanh_layer.hpp, src/caffe/layers/cudnn_tanh_layer.cpp, src/caffe/layers/cudnn_tanh_layer.cu

2. 用caffe源码中的这个文件替换掉faster rcnn 对应文件

include/caffe/util/cudnn.hpp

 

3. 将 faster rcnn 中的 src/caffe/layers/cudnn_conv_layer.cu 文件中的所有

cudnnConvolutionBackwardData_v3 函数名替换为 cudnnConvolutionBackwardData

cudnnConvolutionBackwardFilter_v3函数名替换为 cudnnConvolutionBackwardFilter

번호 제목 글쓴이 날짜 조회 수
319 넌센스 퀴즈 file WHRIA 2016.09.04 75
318 공유 메모리 예제 WHRIA 2016.09.09 121
317 윈도우 서비스에서 유저 권한으로 createprocess WHRIA 2016.09.10 1996
316 힘든 하루하루 WHRIA 2016.09.30 77
315 youtube WHRIA 2016.09.30 39
314 만기까지 끌어 올림. WHRIA 2016.09.30 70
313 . WHRIA 2016.10.01 81
312 KMA 연수교육 주소 WHRIA 2016.10.05 100
311 sqlite to mysql WHRIA 2016.10.18 57
310 caffe WHRIA 2016.10.19 106
309 이미지 size 줄이는 batch WHRIA 2016.10.19 72
308 이미지를 정사각형으로 만드는 batch WHRIA 2016.10.22 1177
307 % 빼고 center crop WHRIA 2016.10.22 297
306 jquery java WHRIA 2016.12.13 1757
» fast rcnn cudnn 5.0 WHRIA 2017.01.08 128

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소