执行时间效率问题

Huangdebo / Confluence

A Robust Non-IoU Alternative to Non-Maxima Suppression in Object Detection

50 stars 10 forks source link

Open miaohaiwei opened 3 years ago

miaohaiwei commented 3 years ago

我使用yolo的调用函数，发现比pytorch的torchvision.ops.boxes.nms效率要慢很多，在置信度阈值设置为0.001的状态下，一张图执行要5秒多，麻烦问下这个可以怎么解决么？

Huangdebo commented 3 years ago

我使用yolo的调用函数，发现比pytorch的torchvision.ops.boxes.nms效率要慢很多，在置信度阈值设置为0.001的状态下，一张图执行要5秒多，麻烦问下这个可以怎么解决么？

把置信度阈值设大一些，一般物体检测都设到0.1-0.3的，这样可以减少很多候选目标。不过还是比 nms 慢不少。还有需要微调一下那个 p_thres，不然可能效果不太好

gpsherry commented 3 years ago

你这是O(n*(n-1))啊，nms是O(n)

Huangdebo commented 3 years ago

你这是O(n*(n-1))啊，nms是O(n)

这个算法的时间复杂度的确比 nms 的大，我的复现也比较粗糙，所以耗时就更多了