yolov5实现推理时标注框只标注置信度最大
实现此功能只需将以下代码插入for *xyxy, conf, cls in reversed(det):
前即可,注意缩进,和for *xyxy, conf, cls in reversed(det):
同级
# print(len(det))
# print(det[0][4])
# 实现在所有标注框中只标注置信度最大的
MaxConf = []
for i in range(len(det)):
# print(i)
MaxConf.append(det[i][4])
Max = max(MaxConf)
# 最大值
MaxI = MaxConf.index(Max)
det = det[[MaxI]]
其中,det是返回的包含标注信息的矩阵对象,其中包括标注框位置参数,置信度等,其中第五个(从一开始)是置信度,当有多个框时,det的长度等于标注框数目,