Skip to content

OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py", line 41, in obb_nms ori_inds = ori_inds[~too_small] #207

@binfenhuayun

Description

@binfenhuayun

in /home/ym/OBBDetection/mmdet/ops/nms_rotated/nms_rotated_wrapper.py
in Line40 应该添加:
ori_inds = ori_inds.to(too_small.device)
否则会报设备不一致的错误

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions