IrisRainbowNeko/AuroraVision

All in one basic anime CV model. tagger+siglip+natural language

AuroraVision

AuroraVision是一个通用的二次元感知模型。可以对二次元图片进行标注，特征提取，相似度计算(图文，图图)等操作。 后续还会提供目标检测相关功能。

模型结构与ML-Danbooru类似

安装

pip install -r requirements.txt

预训练模型

https://huggingface.co/7eu7d7/AuroraVision

使用

图像打标签

python infer.py --ckpt_path 模型权重路径 --tags_path tag汇总文件路径 --img 图片路径