一种基于对比学习的图像生成方法、装置、设备及介质

    公开(公告)号:CN116050481A

    公开(公告)日:2023-05-02

    申请号:CN202211740313.1

    申请日:2022-12-29

    Abstract: 本发明提供了一种基于对比学习的图像生成方法、装置、设备及介质,所述方法包括:将准备好的图文数据对构建为图文数据集,图文数据对为图文之间呈关联关系的图文数据,利用图文数据集对预先基于对比学习构建的图像生成模型进行训练,以得到训练后的图像生成模型,图像生成模型中包含用于对图文特征进行对齐的对齐模块以及用于对图文特征进行融合生成的生成模块,当获取到待处理的文本数据,则利用训练后的图像生成模型生成与待处理文本数据对应的图像。本方法在图文特征融合生成之前,先使用对比学习对图文特征进行对齐,使生成的图像更加具有多样性,且模型简单易于训练,容易落地实施。

Patent Agency Ranking