DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation
erro with nms_rotate_gpu
Aperta
- Lingua principale
- Python
- Stelle
- 312
- Fork
- 92
- Metriche di merge delle PR
- Nessuna PR unita negli ultimi 30g
Descrizione
我发现当图像中物体较多的时候，用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
请问这个报错是什么原因导致的呢？
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.