图像生成器的训练方法、装置、设备以及存储介质

    公开(公告)号:CN112149634B

    公开(公告)日:2024-05-24

    申请号:CN202011143074.2

    申请日:2020-10-23

    Inventor: 田飞

    Abstract: 本申请实施例公开了图像生成器的训练方法、装置、设备以及存储介质,涉及深度学习、大数据处理、数据增强、计算机视觉等人工智能技术领域。该方法包括:将同一对象的第一模态图像的编码和第二模态图像的编码分别输入至图像生成器的初始模型,得到第一图像和第二图像;将第一图像和第二图像分别输入至目标网络模型,得到第一图像的特征值和第二图像的特征值;基于第一图像的特征值和第二图像的特征值以及目标网络模型的类型,确定损失函数,并调整图像生成器的参数,得到训练完成的图像生成器,从而提供了一种可以生成第一模态图像和/或第二模态图像中的特定模态图像的图像生成器,有助于解决因特定模态图像识别模型的训练样本过少导致的过拟合问题,提高模型的识别精度。

Patent Agency Ranking