DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation
erro with nms_rotate_gpu
Open
- Dominant language
- Python
- Stars
- 312
- Forks
- 92
- PR merge metrics
- No merged PRs in 30d
Description
我发现当图像中物体较多的时候，用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
请问这个报错是什么原因导致的呢？
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.