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.

Sign up for GitHub

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

Jump to bottom

erro with nms_rotate_gpu #55

Open
ZOUYIyi opened this issue Dec 24, 2019 · 2 comments
Open

erro with nms_rotate_gpu #55

ZOUYIyi opened this issue Dec 24, 2019 · 2 comments

Comments

@ZOUYIyi
Copy link

ZOUYIyi commented Dec 24, 2019

我发现当图像中物体较多的时候，用inference.py预测运行时间较长，发现代码中的nms是用cpu做的，但是当修改为GPU时，代码运行报错：
File "rotate_polygon_nms.pyx", line 22, in rotate_polygon_nms.rotate_gpu_nms

IndexError: Out of bounds on buffer access (axis 0)

我修改的地方是libs/detection_operations/proposal_opr_.py Line60 use_gpu=True

请问这个报错是什么原因导致的呢？
@zhouzhubin
Copy link

@ZOUYIyi 你好，请问你解决了吗？我也是有这种错误

@yangxue0827
Copy link
Member

可能是输入的box个数为0导致的

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@yangxue0827 @zhouzhubin @ZOUYIyi