• python 版Faster Rcnn


    直接按照官网https://github.com/rbgirshick/py-faster-rcnn上的教程对faster Rcnn进行编译的时候,会发有一些层由于cudnn版本的更新,会报错如下:

    /cudnn_sigmoid_layer.cu(13): error: argument of type "cudnnActivationMode_t" is incompatible with parameter of type "cudnnActivationDescriptor_t"

    在网上查了一下,是由于cudnn的版本更新,接口不兼容造成的,网上有一些博客上采取了修改接口的方法,但是有好几个cpp以及hpp都会修改,比较麻烦,后来在https://github.com/rbgirshick/py-faster-rcnn/issues/237查到了解决办法:

    cd caffe-fast-rcnn  
    git remote add caffe https://github.com/BVLC/caffe.git  
    git fetch caffe  
    git merge -X theirs caffe/master  
    

    Remove self_.attr("phase") = static_cast<int>(this->phase_); from include/caffe/layers/python_layer.hpp after merging.

    即用目前最新版本的caffe来编译faster rcnn,因为新版本的caffe是采用的cudnn5,因此再按照官网的教程进行编译就不会出错,我机器上是cudnn6,也可以编译通过。

  • 相关阅读:
    java数组
    java 常用类
    java 集合(一)
    mysql相关操作(一)
    记录java BigDecimal
    hxg-yw
    一个困惑我好久的问题
    关于重载和重写的一些小知识
    几个常用的HTTP状态码
    死锁产生的原因和条件简述
  • 原文地址:https://www.cnblogs.com/rainsoul/p/8241403.html
Copyright © 2020-2023  润新知