DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation

erro with nms_rotate_gpu

Offen
#55 2 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
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

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.