ComfyUI Extension: comfyui-lsnet

Authored by spawner1145

Created

Updated

40 stars

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 技术核心,本工具聚焦两大核心场景:

    1. 画风分类:精准识别单幅作品的风格属性(如动漫、写实、水彩、国风等),如同写轮眼一眼看穿事物本质,快速完成风格标签匹配;

    2. 画风聚类:自动对多组作品按风格特征进行归类聚合,筛选出风格相似的作品群体,实现批量风格整理与分析,提升处理效率

    第一步:下载必要文件

    前往 Hugging Face 仓库,下载两个核心文件:

    • best_checkpoint.pthLSNet 模型权重文件,决定画风分类与聚类的精度)

    • 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文件夹

    • lsnet内创建子文件夹:

    创建子文件夹

    2. 安装依赖

    将上述两个文件放入子文件夹后,执行以下命令安装 LSNet 运行所需依赖(webui插件可以跳过这一步,会自动安装依赖):

    pip install -r requirements.txt --upgrade
    

    第三步:启动 ComfyUI 并使用

    1. 按常规方式启动 ComfyUI

    2. 在画风分析工作流中,调用 LSNet 相关节点,即可触发 “画风分类” 或 “画风聚类” 功能

    ps:目前版本该插件已经可以单独启动和作为webui插件启动,单独启动在项目根目录运行python -m scripts.app,模型路径为根目录下models/lsnet文件夹内,webui插件启动和comfyui差不多

    使用示例

    基础推理界面(含LSNet核心节点)

    聚类暂时不作示例,内有其他节点供开发者使用

    致谢

    感谢 @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}, 
    }