py-faster-rcnn 训练常见错误

来源:互联网 发布:实物期权 知乎 编辑:程序博客网 时间:2024/05/29 18:07

py-faster-rcnn 训练常见错误

在训练 py-faster-rcnn 时,常会出现以下错误:

  • TypeError: ‘numpy.float64’ object cannot be interpreted as an index
  • Check failed: error == cudaSuccess (4 vs. 0) unspecified launch failure
  • AttributeError: ‘module’ object has no attribute ‘text_format’

TypeError: ‘numpy.float64’ object cannot be interpreted as an index

  • 解决方法 sudo pip install -U numpy==1.11.0`

Check failed: error == cudaSuccess (4 vs. 0) unspecified launch failure

  • 解决方法
    • ① 安装cudnn:
            cp caffe/inlude/caffe/layers/cudnn_* caffe-fast-rcnn/include/caffe/layers/            cp caffe/src/caffe/layers/cudnn_* caffe-fast-rcnn/src/caffe/layers/            cp caffe/include/caffe/util/cudnn.hpp caffe-fast-rcnn/include/caffe/util/            cd caffe-fast-rcnn            make clean            make -j8            make pycaffe
 - ② 修改配置文件 py-faster-rcnn/lib/fast_rcnn/config.py:将BATCH_SIZE调小,比如64,32,16,8....
        # Minibatch size (number of regions of interest [ROIs])        __C.TRAIN.BATCH_SIZE = 128

AttributeError: ‘module’ object has no attribute ‘text_format’

  • 在py-faster-rcnn/lib/fast_rcnn/train.py中添加: import google.protobuf.text_format
原创粉丝点击