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