DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation
erro with nms_rotate_gpu
- Vorherrschende Sprache
- Python
- Sterne
- 312
- Forks
- 92
- PR-Merge-Kennzahlen
- Keine gemergten PRs in 30 T.
Beschreibung
我发现当图像中物体较多的时候,用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
请问这个报错是什么原因导致的呢?
Beitragsleitfaden
Für dieses Repository ist kein Beitragsleitfaden indexiert
Rechercherichtung
Beginne damit, den Fehler aus inference.py zu reproduzieren, nachdem du in Zeile 60 von libs/detection_operations/proposal_opr_.py use_gpu=True gesetzt hast. Lies rotate_polygon_nms.pyx im Umfeld von rotate_gpu_nms und vergleiche die Eingabepuffer, die dort ankommen; fertig ist die Aufgabe, wenn der Out-of-Bounds-Fehler behoben ist und GPU NMS für Bilder mit vielen Objekten abgeschlossen wird.
Vom Indexierungsmodell aus dem Issue-Text verfasst.
Bewertung
- Tech-Stack
- python, tensorflow
- Bereich
- computer-vision, machine-learning
- Issue-Typ
- Bug
- Schwierigkeit
- 4/5
- Geschätzter Aufwand
- 3-5 Tage
- Aktivitätsstatus
- Veraltet
- Klarheit
- Muss geklärt werden
- Anfängerfreundlichkeit
- 30/100