基于扩散模型的前端界面生成方法、系统、介质及终端
摘要:
本发明提供了一种基于扩散模型的前端界面生成方法、系统、介质及终端,包括:获取图片训练数据;对所述图片训练数据进行预处理,得到对应的文字训练数据;根据所述图片训练数据和文字训练数据,对扩散模型进行训练,得到第一扩散模型,并通过所述第一扩散模型生成前端图片初稿和前端线框图初稿;通过分割模型将所述前端图片初稿和所述前端线框图初稿进行融合,得到组合后的前端图片;根据所述第一扩散模型将所述组合后的前端图片进一步融合训练,得到最终的前端界面。本发明使用互联网爬取并标注的网页图片作为训练数据,微调了原模型,在前端生成的任务上表现更为优异。
0/0