基于 Stable Diffusion 与 ControlNet 的 JieHua 画作风格特征提取模型
人工智能
2024-08-22 v1 计算机视觉与模式识别
摘要
本研究提出一种 novel 方法,用于提取 Jiehua(借画)的风格特征:利用基于 ControlNet 的微调 Stable Diffusion 模型(FSDMC)来细化来自艺术家 Jiehua 的描绘技法。FSDMC 的训练数据基于从互联网收集的开源 Jiehua 艺术家作品,随后以(原始图像、Canny 边缘特征、文本提示)格式进行人工构建。通过本文确定的最优超参数,观察到 FSDMC 在数据集上实现 FID 为 3.27,并且在专家评估方面超过 CycleGAN。这不仅表明该模型在提取 Jiehua 风格特征方面效果卓越,同时保留了原始预训练语义信息。本研究的发现表明,在合适的超参数下,FSDMC 可提高 Stable Diffusion 模型在传统艺术风格迁移任务中的效果,特别是在 Jiehua 语境中。
引用
@article{arxiv.2408.11744,
title = {JieHua Paintings Style Feature Extracting Model using Stable Diffusion with ControlNet},
author = {Yujia Gu and Haofeng Li and Xinyu Fang and Zihan Peng and Yinan Peng},
journal= {arXiv preprint arXiv:2408.11744},
year = {2024}
}
备注
accepted by ICCSMT 2024