目标检测-检测结果按类别置信度排序

一、前言：

在做目标检测任务时，检测器输出的检测结果往往需要进行处理，比如：

对于单类别检测模型（如检测一个工件中一个圆孔，单个工件中只有1个孔），那么业务逻辑中就需要对检测框按置信度排序。

二、实现：

#usr/bin/env python
#-*- coding: utf-8 -*-


det_results = []
det_1 = [0.99,100.1,200.2,300.3,400.4,0]
det_2 = [0.88,400.1,500.2,600.3,700.4,0]
det_3 = [0.77,800.1,900.2,100.3,200.4,0]

det_results.append(det_1)
det_results.append(det_2)
det_results.append(det_3)

det_results.sort(key = lambda det: (det[0]))  # 按需要的数据索引排序

print(det_results)