yolov5实现推理时标注框只标注置信度最大

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的长度等于标注框数目,

你可能感兴趣的:(python,人工智能,目标检测)