ComfyUI Extension: comfyui-lsnet
A nodepack powered by LSNet that classifies and clusters artworks by style for fast and efficient style analysis.
Custom Nodes (0)
README
comfyui-lsnet
「我这双眼睛,能将黑暗看得一清二楚」
—— 宇智波佐助
“Kaloscope”(万花筒)致敬万花筒写轮眼,象征忍术(画风)复刻能力
该插件支持comfyui插件,webui插件和单独启动三种方式运行
核心能力
基于 LSNet 技术核心,本工具聚焦两大核心场景:
-
画风分类:精准识别单幅作品的风格属性(如动漫、写实、水彩、国风等),如同写轮眼一眼看穿事物本质,快速完成风格标签匹配;
-
画风聚类:自动对多组作品按风格特征进行归类聚合,筛选出风格相似的作品群体,实现批量风格整理与分析,提升处理效率
第一步:下载必要文件
前往 Hugging Face 仓库,下载两个核心文件:
-
best_checkpoint.pth
(LSNet 模型权重文件,决定画风分类与聚类的精度) -
class_mapping.csv
(风格类别映射配置文件,适配不同画风标签的识别与归类)
huggingface仓库地址:https://huggingface.co/heathcliff01/Kaloscope/tree/main
或者在modelscope下载:https://www.modelscope.cn/models/Heathcliff02/Kaloscope/files
第二步:文件放置与环境配置
1. 创建目录结构
在 ComfyUI 的models
目录下,新建名为lsnet
的文件夹(专门存放 LSNet 相关模型文件);
进入lsnet
文件夹后,可随意创建一个子文件夹(如 “checkpoints”“kaloscope” 等,名称无强制要求,用于归类核心文件)。
目录结构示例:
ComfyUI/
└── models/
└── lsnet/
└── 子文件夹名称/ # 例:"sharingan" 或 “kaloscope”
├── best_checkpoint.pth
└── class_mapping.csv
相关操作截图:
- 新建
lsnet
文件夹:
- 在
lsnet
内创建子文件夹:
2. 安装依赖
将上述两个文件放入子文件夹后,执行以下命令安装 LSNet 运行所需依赖(webui插件可以跳过这一步,会自动安装依赖):
pip install -r requirements.txt --upgrade
第三步:启动 ComfyUI 并使用
-
按常规方式启动 ComfyUI
-
在画风分析工作流中,调用 LSNet 相关节点,即可触发 “画风分类” 或 “画风聚类” 功能
ps:目前版本该插件已经可以单独启动和作为webui插件启动,单独启动在项目根目录运行python -m scripts.app,模型路径为根目录下models/lsnet文件夹内,webui插件启动和comfyui差不多
使用示例
聚类暂时不作示例,内有其他节点供开发者使用
致谢
感谢 @heathcliff01 训练模型
训练代码
https://github.com/spawner1145/lsnet-test.git
Citation
@misc{wang2025lsnetlargefocussmall,
title={LSNet: See Large, Focus Small},
author={Ao Wang and Hui Chen and Zijia Lin and Jungong Han and Guiguang Ding},
year={2025},
eprint={2503.23135},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2503.23135},
}