这里采用https://blog.csdn.net/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.csdn.net/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就不会报错了