ComfyUI-Impact-Pack>SEGS Filter (non max suppression)
ComfyUI-Impact-Pack

SEGS Filter (non max suppression)

节点功能:该节点使用非极大值抑制(NMS)算法来过滤重叠的检测区域,只保留置信度最高的检测结果。
SEGS Filter (non max suppression)-节点参数说明
输入参数
segs输入的分割区域集合。
输出参数
filtered_SEGS执行非极大值抑制后保留的区域集合。
控件参数
iou_thresholdIoU(交并比)阈值,超过该值即认为重复。
NMS 过滤逻辑如下:
  1. 按置信度从高到低排序所有检测区域
  2. 选择置信度最高的区域并保留
  3. 计算该区域与剩余区域的 IoU
  4. 移除 IoU 超过阈值的重叠区域
  5. 重复此过程直到处理完所有区域

它主要是针对检测的同一个目标进行处理,比如图中只有一张人脸,却出现多个检测框,因此需要筛选出最精准的一个检测框。

如下,红框中的这两个bbox相近,通过nms中的iou_threshold后,就有一张被筛除出去了。

广告

可加入星球获取所有示例工作流~

广告

微信扫码入群,加入AIGC大家庭吧!

如您有任何疑问和网站建议,也欢迎加入我们的QQ交流群!

QQ群:128763578