No module named gpu_nms

[Tclz]

Hi, when i run the generate_tsv.py, i meet the following problem

Traceback (most recent call last): File "./tools/generate_tsv.py", line 26, in from fast_rcnn.test import im_detect,_get_blobs File "/home/vilbert_beta/bottom-up-attention/tools/../lib/fast_rcnn/test.py", line 17, in from fast_rcnn.nms_wrapper import nms, soft_nms File "/home/vilbert_beta/bottom-up-attention/tools/../lib/fast_rcnn/nms_wrapper.py", line 9, in from nms.gpu_nms import gpu_nms ImportError: No module named gpu_nms

What should i do to fix it ? plz help me, thanks in advance! :)

[alice-cool]

your caffe environment is not installed completely