yolov5中NMS
时间: 2023-10-02 15:05:05 浏览: 87
基于yolov5的旋转目标检测yolov5-obb-master.zip
在YOLOv5中,NMS(非极大值抑制)是一种用于去除冗余边界框的技术。NMS的实现在代码中的这一部分:c = x[:, 5:6] * (0 if agnostic else max_wh) # classes,其中x[:, 5:6]表示类IDX,而max_wh是一个偏移量。这里的agnostic参数决定了NMS是对多个类别一起进行计算还是按照不同的类别分别计算NMS,当agnostic为True时,表示多个类别一起计算NMS,而当agnostic为False时,表示按照不同的类别分别计算NMS。通过NMS,可以根据预测框的得分和重叠度,选择最具有代表性的边界框作为最终的检测结果。
阅读全文