Skip to content

erro with nms_rotate_gpu #55

@ZOUYIyi

Description

@ZOUYIyi

我发现当图像中物体较多的时候,用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

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions