erro with nms_rotate_gpu

[ZOUYIyi]

我发现当图像中物体较多的时候，用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]

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

[yangxue0827]

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