Skip to content

fast rcnn cudnn 5.0

2017.01.08 15:47

WHRIA 조회 수:157

手动修改文件,参考了卜居大神的博客 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

번호 제목 글쓴이 날짜 조회 수
1519 asyncio WHRIA 2020.07.23 308
1518 good luck ! WHRIA 2019.03.21 313
1517 민원24 프린터제한을 우회하거나 PDF 로 저장하는 방법 file WHRIA 2015.07.12 315
1516 IEEE Spectrum WHRIA 2018.02.27 328
1515 IPTV WHRIA 2015.11.01 329
1514 % 빼고 center crop WHRIA 2016.10.22 329
1513 reverse VNC WHRIA 2018.02.14 345
1512 python pdf writer WHRIA 2019.11.04 348
1511 CNN Models WHRIA 2018.12.23 352
1510 scopus [1] WHRIA 2020.08.02 352
1509 Keratinocytic Skin Cancer Detection on the Face Using Region-Based Convolutional Neural Network WHRIA 2019.12.07 358
1508 미래가 보이지 않는다. WHRIA 2017.07.01 365
1507 gpt2 WHRIA 2023.02.21 394
1506 Windows Subsystem Linux WHRIA 2018.01.01 424
1505 LG 21 세기 brain hunt - USA file WHRIA 2019.05.23 434

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소