[发明专利]基于对抗生成网络的数据增广方法、装置、计算机及介质

权利要求书

1.一种基于对抗生成网络的数据增广的方法，其特征在于，包括：

获取目标域中的第一图片与源域中的第二图片；

将所述第一图片输入至第一生成网络，得到伪第二图片；将所述伪第二图片输入至第二生成网络，得到伪第一图片；所述第一生成网络与所述第二生成网络互为相反的网络；

计算所述第一图片与所述伪第一图片的一致性约束值；

将所述第二图片与所述伪第二图片输入至判别网络，得到所述判别网络对所述第二图片及所述伪第二图片的分类结果；

根据损失函数计算所述分类结果的损失值；

若所述一致性约束值与所述损失值的和满足预设值，将所述伪第一图片添加至目标域中，以增加目标域中的图片数据。

2.根据权利要求1所述的基于对抗生成网络的数据增广的方法，其特征在于，所述获取目标域中的第一图片与源域中的第二图片之后，还包括：

将所述第二图片输入至第二生成网络，得到假第一图片；将所述假第一图片输入至第一生成网络，得到假第二图片；

计算所述第二图片与所述假第二图片的候选一致性约束值；

将所述第二图片与所述假第二图片输入至第二判别网络，得到所述第二判别网络对所述第二图片及所述假第二图片的候选分类结果；

根据损失函数计算所述候选分类结果的候选损失值；

若所述候选一致性约束值与所述候选损失值的和满足预设值，将所述假第二图片添加至源域中，以增加源域中的图片数据。

3.根据权利要求1或2所述的基于对抗生成网络的数据增广的方法，其特征在于，所述源域与目标域为相互映射的数据集。

4.根据权利要求1所述的基于对抗生成网络的数据增广的方法，其特征在于，所述若所述一致性约束值与所述损失值的和满足预设值，将所述伪第一图片添加至目标域中之前，包括：

获取当前数据增广的执行阶段；

根据所述执行阶段匹配一致性约束值与所述损失值之和的预设值。

5.根据权利要求1所述的基于对抗生成网络的数据增广的方法，其特征在于，所述获取目标域中的第一图片与源域中的第二图片之前，还包括：

从若干个待选域中选取一个作为目标域，获取所述目标域的图片等级；

根据预设的等级规则匹配与所述目标域的图片等级相邻的候选图片等级；

根据所述候选图片等级确定所述目标域对应的源域。

6.根据权利要求5所述的基于对抗生成网络的数据增广的方法，其特征在于，所述从若干个待选域中选取一个作为目标域，包括：

获取各个待选域的图片数量；

根据所述图片数量对所述各个待选域进行排序；

从排序后的待选域中，根据图片数量从多至少依次选取对应的待选域作为目标域。

7.根据权利要求5所述的基于对抗生成网络的数据增广的方法，其特征在于，所述目标域包括重度黑眼圈图片等级与轻度黑眼圈图片等级的图片，所述源域包括轻度黑眼圈图片等级与无黑眼圈图片等级的图片；所述根据预设的等级规则匹配与所述目标域的图片等级相邻的候选图片等级，包括：

若所述候选图片等级为重度黑眼圈图片等级，确定所述目标域对应的源域的图片等级为轻度黑眼圈图片等级；

若所述候选图片等级为轻度黑眼圈图片等级，确定所述目标域对应的源域的图片等级为无黑眼圈图片等级。