一种基于迁移学习及生成对抗网络的图像数据扩增方法

文档序号:34181044发布日期:2023-05-17 08:54阅读:来源:国知局

技术特征:

1.一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,包括以下步骤:

2.根据权利要求1所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,所述融合包括以下步骤:

3.根据权利要求2所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,对第三数据集进行预处理的步骤包括:

4.根据权利要求3所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,第一编码器的训练方式为监督训练与对抗训练相结合,第一向量判别器的训练方式为对抗训练;

5.根据权利要求4所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,用于对第一编码器进行训练的监督训练的损失函数为l2损失函数,具体形式为:

6.根据权利要求5所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,用于对第一编码器以及第一向量判别器进行对抗训练的损失函数具体形式为:

7.根据权利要求6所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,所述gan网络模型包括修饰器和第二gan,第四数据集上进行gan网络模型训练的步骤包括:

8.根据权利要求7所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,修饰器以及第二图像生成器的前向传播过程包括以下步骤:

9.根据权利要求8所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,用于对修饰器、第二图像生成器以及第二图像判别器进行联合对抗训练的损失函数具体形式为:

10.根据权利要求9所述的一种基于迁移学习及生成对抗网络的图像数据扩增方法,其特征在于,利用训练完成后的修饰器和第二图像生成器对目标数据集进行数据扩增的步骤包括:


技术总结
本发明公开一种基于迁移学习及生成对抗网络的图像数据扩增方法,包括以下步骤:利用GAN的第一图像生成器收集输入的隐向量与输出图像的样本对数据集,称为第一数据集;把第一数据集中的每一对数据融合,得到第二数据集;构建第一编码器,其输入、输出格式分别与第一数据集的图像、隐向量相同;构建第一向量判别器,其输入格式与第二数据集的数据相同;利用第一数据集以及第二数据集,对第一编码器以及第一向量判别器进行训练;收集第三数据集;对第三数据集预处理获得第四数据集;将第四数据集的图像依次输入训练后的第一编码器,得到相同数量的隐向量数据集,称为第五数据集;对GAN网络模型进行训练;利用训练后GAN网络模型进行数据扩增。

技术研发人员:周智恒,李志豪,陶希远,曹英烈,杨俊怡
受保护的技术使用者:华南理工大学
技术研发日:
技术公布日:2024/1/12
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1