search

YuwenXiong / py-R-FCN

R-FCN with joint training and python support
MIT License
1.05k stars 472 forks source link

demo,demo_rfcn use cpu version of nms when in cpu_mode #87

Open samlich opened 7 years ago

samlich commented 7 years ago

nms_wrapper.py uses CPU mode when cfg.USE_GPU_NMS isn't set or it's passed force_cpu as True.

When --cpu was provided, demo.py and demo_rfcn.py did not set cfg.USE_GPU_NMS, which defaulted to True. So, NMS always operated in GPU mode. This caused sometimes the console to spew 'unknown error', and the bounding boxes on each object to not be combined; in other setups the program did not run at all.