ComfyUI Extension: comfyui_imgutils
Image tool kit for comfyui with dghs-imgutils
Custom Nodes (0)
README
comfyui_imgutils
这个是 ComfyUI 的图像处理工具集,提供了多种图像处理和检测功能
部分功能基于imgutils库
https://github.com/deepghs/imgutils
模型文件全部由该库提供和管理下载,会自动下载到
HF_HOME
环境变量指定的目录下
因此需要在 ComfyUI 的启动项中设置该环境变量
Segment-Anything 模型需要手动下载到
ComfyUI\models\sams
目录下(impact-pack 同款路径)
由于我没研究明白的原因,该路径不受extra_model_paths.yaml
的影响(如果你有解决方案欢迎 PR)
============================= LK ==================================
节点介绍:
检测节点
-
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
- 用于
Mask
和BBox
之间的转换
- 用于
-
BBoxFilter
- 用于过滤
BBox
,可以根据置信度、面积和 标签进行过滤labels
的值依照的是Imgutils Generic Detector
输出的图片中bbox上标注的标签,可以输入多个,用逗号分隔
- 用于过滤
segment-anything节点
基本上抄 Impact-Pack 的
segment 节点
-
Imgutils Auto Segmenter
- 仅能对图片进行 前景和背景 的分割
-
Imgutils BBox Segmenter
- 在
Imgutils Generic Detector
的基础上,使用检测到的BBox
进行分割
- 在
打码节点
- Censor with Mask
- 利用提供的
Mask
对图片进行打码处理censor_type
:打码类型blur
:模糊pixelate
:像素化color
:纯色遮挡
- 利用提供的
Mask处理节点
收录一些常用的 Mask 处理节点
-
Mask Morphology:
- 提供了常用的形态学操作:膨胀、腐蚀、开运算和闭运算
-
Mask Edge Operations:
- 提供了常用的边缘处理操作::扩展、收缩、扩展收缩和收缩扩展
-
Mask Attributes:
- 提供了常用的属性调整操作:二值化、阈值处理、平滑处理、对比度和亮度调整、反转和二值化
-
Mask Combine:
- 提供了多mask合并操作:加法、减法、乘法、除法、最大值、最小值和异或操作
-
Mask Info:
- 显示mask的统计信息,如形状、覆盖率、范围和均值
-
MaskHelperLK:
- 如果你忘记了以上
Mask
处理节点的功能,可以使用这个节点查看,因为我知道这个东西只看节点名字很难知道效果
- 如果你忘记了以上
懒了,今天先写到这,传播民主与自由去了。