DetectionTeamUCAS / DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation
erro with nms_rotate_gpu
- Lenguaje dominante
- Python
- Estrellas
- 312
- Forks
- 92
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
我发现当图像中物体较多的时候,用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
请问这个报错是什么原因导致的呢?
Guía de contribución
No hay ninguna guía de contribución indexada para este repositorio
Línea de trabajo
Empieza reproduciendo el fallo de inference.py después de establecer use_gpu=True en la línea 60 de libs/detection_operations/proposal_opr_.py. Lee rotate_polygon_nms.pyx alrededor de rotate_gpu_nms y compara los búferes de entrada que llegan a esta función; se considera terminado cuando se resuelva el error de acceso fuera de límites y GPU NMS se complete para imágenes que contengan muchos objetos.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Stack tecnológico
- python, tensorflow
- Área
- computer-vision, machine-learning
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Necesita aclaración
- Aptitud para principiantes
- 30/100