DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation

erro with nms_rotate_gpu

Ouverte
#55 2 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
Langage dominant
Python
Étoiles
312
Forks
92
Métriques de merge des PR
Aucune PR mergée en 30 j

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

请问这个报错是什么原因导致的呢?

Guide de contribution

Aucun guide de contribution indexé pour ce dépôt

Piste de recherche

Commencez par reproduire l’échec de inference.py après avoir défini use_gpu=True à la ligne 60 de libs/detection_operations/proposal_opr_.py. Lisez rotate_polygon_nms.pyx autour de rotate_gpu_nms et comparez les buffers d’entrée qui lui parviennent ; le travail est terminé lorsque l’erreur de dépassement de limites est résolue et que GPU NMS s’exécute jusqu’au bout pour les images contenant de nombreux objets.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, tensorflow
Domaine
computer-vision, machine-learning
Type d'issue
Bug
Difficulté
4/5
Temps estimé
3-5 jours
Activité
À l'abandon
Clarté
À clarifier
Accessibilité débutants
30/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.