Yolov5设置不同类别元素之间使用NMS非极大值抑制

Yolov5默认设置类间元素分别使用NMS,这样一来,会使得图像中的相近区域检测出多个类别目标。
如果想要所有的元素一起使用NMS,可以这样设置:
进入utils的general.py,在791行添加

    agnostic = True

Yolov5设置不同类别元素之间使用NMS非极大值抑制_第1张图片

进行类间NMS时

改为True后:

你可能感兴趣的:(yolov5实战,计算机视觉,人工智能,深度学习)