可在线运行的notebook:在kaggle网站搜索
首先需要安装依赖:
接着加载初始图像:
将初始图像转换为姿态图像:
然后加载文生图模型:(其中注释部分可在CPU充裕时取消注释,它可以将部分GPU负载迁移到CPU)
同时设置提示词,用于指导模型绘图:
最终将底图和提示词传入模型,绘制图像: