Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

gpu_nms does not work. #939

Open
zxGuo opened this issue Jul 15, 2020 · 2 comments
Open

gpu_nms does not work. #939

zxGuo opened this issue Jul 15, 2020 · 2 comments

Comments

@zxGuo
Copy link

zxGuo commented Jul 15, 2020

Recently I reinstalled caffe and py-faster-rcnn, everythong works well except that I get a lot of overlapped detection bounding boxes in my result. Then I find it is because the gpu_nms always outputs all the detection results without doing anything, while cpu_nms works well.

Then I tried to track the problem and I found in 'nms_kernel.cu', the function 'nms_kernel' is skipped.

_ global _ void nms_kernel(const int n_boxes, const float nms_overlap_thresh,
const float *dev_boxes, unsigned long long *dev_mask)

Since I do not know much about cuda programming, I can not go further.
Can someone help me with this problem? Thanks in advance.

@zxGuo
Copy link
Author

zxGuo commented Jul 15, 2020

My environment is RTX 2080, ubuntu 18.04, python 3.7, cuda 10.0

@RizhaoCai
Copy link

The

My environment is RTX 2080, ubuntu 18.04, python 3.7, cuda 10.0

The problems maybe because of python3.7.
python3.5 may help.

Btw, do you manage to solve the problem with python3.7

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants