Usage of poly_nms_gpu

[DrRyanHuang]

Hi, thx for your very helpful code.
Could you please give a demo of poly_nms_gpu?

[yoloyash]

Hello, can you explain how to use it please?

[dingjiansw101]

The function can be found at this link
https://github.com/CAPTAIN-WHU/DOTA_devkit/blob/master/poly_nms_gpu/nms_wrapper.py
The interface of poly_nms_gpu is the same as py_cpu_nms_poly https://github.com/CAPTAIN-WHU/DOTA_devkit/blob/99388551054be9a6dabb01c8bb2a7eb562d57b4f/ResultMerge.py#LL17C1-L17C1

[yoloyash]

@dingjiansw101 Thanks a lot for your reply. It's working now. However, it's very slow for me. Is there anything I'm doing wrong or missing out? Trying to follow yolov5_obb implementation and the nms per batch exceeds 50-60 seconds in some cases. I'm using it as follows. [cpu:Intel(R) Xeon(R) CPU E5-1650 v3 @ 3.50GHz, gpu: Nvidia Tesla T4]
from poly_nms_gpu import poly_nms
poly_nms.poly_gpu_nms(dets, thresh=0.5, device_id=0)

[dingjiansw101]

Actually, I recommend you use py_cpu_nms_poly. GPU version is not always faster.