Extensions/comfyui_imgutils
ComfyUI Extension

comfyui_imgutils

Image tool kit for comfyui with dghs-imgutils

By LK-168·Created about a year ago·Updated about a year ago· 1
LK-168/comfyui_imgutils
Nodes14
On cloudLocal install
Categoryimgutils/bbox, imgutils/censor
Stars1
Updatedabout a year ago
Readme

comfyui_imgutils

这个是 ComfyUI 的图像处理工具集,提供了多种图像处理和检测功能

部分功能基于imgutils库 https://github.com/deepghs/imgutils 模型文件全部由该库提供和管理下载,会自动下载到 HF_HOME 环境变量指定的目录下

因此需要在 ComfyUI 的启动项中设置该环境变量 如果你是官方包,可以在 .\run_nvidia_gpu.bat 的开头中添加:

set "CURDIR=%cd%"
set HF_HOME="%CURDIR%\(你想要的字目录)"

秋叶包可以不用设,默认是下载到 .cache\huggingface\hub\ 下面

Segment-Anything 模型需要手动下载到 ComfyUI\models\sams目录下(impact-pack 同款路径)

由于我没研究明白的原因,该路径不受extra_model_paths.yaml的影响(如果你有解决方案欢迎 PR)

============================= LK ==================================

节点介绍:

BBox节点

alt text

  • Imgutils Generic Detector

    • 支持多种 imgutils 提供的多种基于anime的检测模型
      • detection_type:检测类型
      • conf_threshold:置信度阈值
      • iou_threshold:IOU阈值, 用于非极大值抑制,总之就是用于合并重叠的检测框的
      • draw_boxes: 是否在输出的图片上绘制检测框
      • level: 模型的类型 n (nano) 和 s (standard)两种
      • version: 模型的版本,不填是用默认的,我之后会改的好一点
  • Mask to BBox 、 BBox to Mask

    • 用于 MaskBBox 之间的转换
  • BBoxFilter

    • 用于过滤 BBox,可以根据置信度、面积和标签进行过滤
      • labels的值依照的是 Imgutils Generic Detector 的输出 image with boxes 中bbox上标注的标签,可以输入多个,用逗号分隔

segment-anything节点

基本上抄的 Impact-Pack

alt text

  • SAMPredictorNode
  • SAMLoader for SAMPredictorNode

segment 节点

alt text

  • Imgutils Auto Segmenter

    • 仅能对图片进行 前景和背景 的分割
  • Imgutils BBox Segmenter

    • Imgutils Generic Detector 的基础上,使用检测到的 BBox 进行分割
打码节点
  • Censor with Mask
    • 利用提供的 Mask 对图片进行打码处理
      • censor_type:打码类型
        • blur:模糊
        • pixelate:像素化
        • color:纯色遮挡

Mask处理节点

收录一些常用的 Mask 处理节点 alt text

  • Mask Morphology:

    • 提供了常用的形态学操作:膨胀、腐蚀、开运算和闭运算
  • Mask Edge Operations:

    • 提供了常用的边缘处理操作::扩展、收缩、扩展收缩和收缩扩展
  • Mask Attributes:

    • 提供了常用的属性调整操作:二值化、阈值处理、平滑处理、对比度和亮度调整、反转和二值化
  • Mask Combine:

    • 提供了多mask合并操作:加法、减法、乘法、除法、最大值、最小值和异或操作
  • Mask Info:

    • 显示mask的统计信息,如形状、覆盖率、范围和均值
  • MaskHelperLK:

    • 如果你忘记了以上Mask处理节点的功能,可以使用这个节点查看,因为我知道以上四个节点只看节点名字很难知道效果