Linux TensorFlow-CPU faster rcnn 踩过的坑

这里采用https://blog.youkuaiyun.com/sinat_33486980/article/details/81045315这个作者的前半部分。非常好用，在此非常感谢

但最后一步运行../tools/demo.py 会报错

File"../tools/demo.py",,line 153, in <module>

              demo(sess, net, im_name)

File "../tools/demo.py", line 98 in demo

       keep=nms(debs, NUMS_THRESH)

File "/home/wsy/tf-faster-rcnn/tools/../lib/model/nms_wrapper.py", line 21,in

nus

      return gpu_nms(dets, thresh, device_id=0)

NameError；global name'gpu_num' is not defind

这里可以想到我们按照https://blog.youkuaiyun.com/sinat_33486980/article/details/81045315里介绍的已经吧GPU设置为了CPU，这里的问题还是没有改完全，因此 进入lib/model/nms_wrapper/py 把报错的地方全都注释掉就可以了，即加“#”就可以.如下所示。

 # if cfg.USE_GPU_NMS and not force_cpu:

#        return gpu_nms(dets, thresh, device_id=0)

在运行demo.py就不会报错了