执行时间效率问题

[miaohaiwei]

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

[Huangdebo]

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

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

[gpsherry]

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

[Huangdebo]

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

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