ComfyUI-ImageSimilarity

项目简介

ComfyUI-ImageSimilarity 是一个为 ComfyUI 设计的自定义节点插件，提供基于感知哈希（pHash）算法的图像相似性检测功能。你可以通过拖拽节点，快速判断两张图片是否相似。

功能特性

• 支持两张图片的相似性对比
• 可自定义相似性阈值（汉明距离）
• 结果以布尔值输出，便于流程控制

安装方法

1. 克隆或下载本仓库到 ComfyUI 的 custom_nodes 目录下：

   git clone https://github.com/你的用户名/ComfyUI-ImageSimilarity.git

2. 安装依赖（在本插件目录下执行）：

   pip install -r requirements.txt

使用说明

1. 启动 ComfyUI。
2. 在节点菜单中找到 Image/Utils 分类下的 Image Similarity Checker 节点。
3. 连接两张待比较的图片，并设置相似性阈值（默认10，范围0-64）。
4. 节点输出布尔值，表示图片是否相似。

依赖项

• torch
• numpy
• pillow
• imagehash>=4.0

其中 torch、numpy、pillow 通常已由 ComfyUI 安装。

节点开发参考

• 节点主逻辑位于 nodes.py 文件，类名为 ImageSimilarityChecker。
• 节点注册与显示名称映射见 __init__.py。

许可证

本项目遵循 MIT License。