faster-rcnn

在没有gpu的机器上运行的时候

lib/fast_rcnn/nms_wrapper.py

#from nms.gpu_nms import gpu_nms

需要注释掉,因为gpu_nms加载了libcudart.so.8.0

另外在运行的时候要首先设置:

caffe.set_mode_cpu()
cfg.USE_GPU_NMS = False

 

之前alexnet样本图片的备忘:

样本必须是随机的,否则loss降不下来

 样本图片在生成lmdb会resize为256*256

 

'numpy.float64' object cannot be interpreted as an index 遇到这个问题,是由于numpy版本1.12引起的,降级就可以。参见https://github.com/rbgirshick/py-faster-rcnn/issues/481

 

 

posted @ 2017-01-18 10:57  怎么也得过啊  阅读(477)  评论(0编辑  收藏  举报