朋友做的美学评分模型:
针对XL训练研发的开源美学评分与数据清洗工具组,可用于大数据集训练的预处理,美学评分与质量词自动标注,目前Anime Aesthetic Variant部分已经在hugging face space开源,如果对大家有用的话,欢迎给项目添一个项目链接为:Anime Thetic - a Hugging Face Space by Laxhar
数据集来自20万张人工挑选的动漫图片,评分维度包括图片质量和构图质量。使用双模型监督架构(DMSA)进行模型评分,
模型参数量:1.1b+110m 画面+构图 vit+grn, 评分范围:
-1~1;
质量提示词分为5个等级;
模型优势:
(1)识别效果更好:针对动漫图片特化的美学评价,可以快速筛选高质量图片;
(2)识别速度更快:在标准环境下,AAV 模型可在 30 分钟内对 10,000 张图片进行高精度评估,对 100 万张动漫图片进行评分也只需 50 小时!
(3)部署简易:模型及源码已在HF开源,您可以针对自身需求进行魔改与微调。您可以通过 HFspace 中的 "anime-thetic"实际体验AAV美学评分模型,HFspace 中提供了基于不同评分标准的示例,也可以使用自己上传的图片。