文章链接:文章链接
源代码:Github
本项目旨在Windows 10 上实现InST的部署工作。原版是Jupyter Notebook的。
其实很简单,可以直接下载我做好了的程序,接下来您自己下一下预训练模型放入Models文件中就行了。
下载链接:优快云除预训练模型外的完整代码;Github下载
接下来讲讲和源代码具体的区别:
1.每次只能运行单张图像,输出也是
2.每次运行时,记得删除Output里的原始图像
文件结构:
文件目录中;Logs是四类模型。分别对应文章中预训练的四种风格;Output会生成图像
重点只看run.py文件
你可以更改的只有下面的model部分,在logs里面选择使用的模型。
Prompt不用管;content_dir是你输入图像;Style_die是你学习的风格图像;ddim不管;strengh是学习的强度,原文为0.6;seed原文为50;
接下来直接运行即可