py-R-FCN icon indicating copy to clipboard operation
py-R-FCN copied to clipboard

demo,demo_rfcn use cpu version of nms when in cpu_mode

Open samlich opened this issue 8 years ago • 0 comments

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.

samlich avatar Oct 15 '17 22:10 samlich