非严格对齐的小样本不同源图像转换方法、系统及设备

文档序号:31606965发布日期:2022-09-21 11:02阅读:来源:国知局

技术特征:
1.一种非严格对齐的小样本不同源图像转换方法,其特征在于,所述图像转换方法包括:步骤s10,获取一个或少量x类别图像;步骤s20,通过训练好的图像转换模型进行所述x类别图像的非对齐转换,获得y类别图像;所述非对齐代表所述图像转换模型在训练阶段,图像中的场景和内容不同;其中,所述图像转换模型基于cyclegan网络构建,所述图像转换模型包括两个呈现对偶关系的生成器和两个判别器,所述生成器基于编码器、转换器和解码器构建,所述判别器基于卷积神经网络和激活函数构建,在所述转换器的resne block模块中添加空间注意力机制。2.根据权利要求1所述的非严格对齐的小样本不同源图像转换方法,其特征在于,所述图像转换模型,其训练方法为:步骤b10,获取不同场景目标的x类别和y类别图像对序列;步骤b20,进行所述x类别和y类别图像对序列中各图像对的非严格对齐,获得第一x类别和y类别图像对序列;步骤b30,将所述第一x类别和y类别图像对序列中各图像的尺寸调整为设定大小,获得第二x类别和y类别图像对序列;步骤b40,分别进行所述第二x类别和y类别图像对序列中各图像的平移、缩放、裁剪的几何变换以及亮度调整、添加高斯噪声操作,获得增强x类别和y类别图像对序列;步骤b50,通过所述增强x类别和y类别图像对序列进行所述图像转换模型的迭代训练,直至所述图像转换模型的总优化损失函数值低于设定阈值或达到设定训练次数,获得训练好的图像转换模型。3.根据权利要求2所述的非严格对齐的小样本不同源图像转换方法,其特征在于,步骤b20中进行所述x类别和y类别图像对序列中各图像对的非严格对齐,其方法为:以各图像对中的目标为基准,将所述x类别和y类别图像对序列中各图像对的成像范围的偏差调整至设定阈值范围内,完成图像的非严格对齐。4.根据权利要求2所述的非严格对齐的小样本不同源图像转换方法,其特征在于,所述总优化损失函数,其表示为:其中,为总优化损失函数,和分别为呈现对偶关系的生成器和的损失函数,和分别为判别器和的损失函数,生成器用于将类别图像转换为类别图像,生成器用于将类别图像转换为类别图像,判别器用于判别生成器基于类别图像转换的类别图像属于类别的概率,判别器用于判别生成器基于类别图像转换的类别图像属于类别的概率,分别为总优化损失函数中损失函数的重要性权重参数。
5.根据权利要求4所述的非严格对齐的小样本不同源图像转换方法,其特征在于,所述生成器,其损失函数表示为:其中,代表当前给定样本,代表转换的目标样本集,代表当前给定样本被判定为类图像的概率,代表生成器将当前给定样本由类图像转换为类图像,代表生成器将生成器转换的图像转换为类图像,代表预设的权重参数,代表图余弦度量函数, (.)表示区域直方图分布函数,代表选定区域的像素集合。6.根据权利要求5所述的非严格对齐的小样本不同源图像转换方法,其特征在于,所述生成器,其损失函数表示为:其中,代表转换后还原的样本集,代表当前给定样本被判定为类图像的概率,代表生成器将当前给定样本由类图像转换为类图像,代表生成器将生成器转换的图像转换为类图像,代表预设的权重参。7.根据权利要求6所述的非严格对齐的小样本不同源图像转换方法,其特征在于,所述判别器和,其损失函数和分别表示为:分别表示为:。8.一种非严格对齐的小样本不同源图像转换系统,其特征在于,所述图像转换系统包括图像获取模块、图像转换模块、模型构建模块、图像对齐模块、图像尺寸调整模块、图像增强模块和模型训练模块;所述图像获取模块,用于获取待转换的一个或少量x类别图像;还用于获取模型训练的不同场景目标的x类别和y类别图像对序列;所述图像转换模块,通过训练好的图像转换模型进行所述x类别图像的非对齐转换,获得y类别图像;所述非对齐代表所述图像转换模型在训练阶段,图像中的场景和内容不同;所述模型构建模块,基于cyclegan网络构建图像转换模型,所述图像转换模型包括两个呈现对偶关系的生成器和两个判别器,并在所述cyclegan网络判别器对应的转换器的resne block模块中添加空间注意力机制,所述生成器基于编码器、转换器和解码器构建,所述判别器基于卷积神经网络和激活函数构建;
所述图像对齐模块,进行所述x类别和y类别图像对序列中各图像对的非严格对齐,获得第一x类别和y类别图像对序列;所述图像尺寸调整模块,将所述第一x类别和y类别图像对序列中各图像的尺寸调整为设定大小,获得第二x类别和y类别图像对序列;所述图像增强模块,分别进行所述第二x类别和y类别图像对序列中各图像的平移、缩放、裁剪的几何变换以及亮度调整、添加高斯噪声操作,获得增强x类别和y类别图像对序列;所述模型训练模块,通过所述增强x类别和y类别图像对序列进行所述图像转换模型的迭代训练,直至所述图像转换模型的总优化损失函数值低于设定阈值或达到设定训练次数,获得训练好的图像转换模型。9.一种电子设备,其特征在于,包括:至少一个处理器;以及与至少一个所述处理器通信连接的存储器;其中,所述存储器存储有可被所述处理器执行的指令,所述指令用于被所述处理器执行以实现权利要求1-7任一项所述的非严格对齐的小样本不同源图像转换方法。10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机指令,所述计算机指令用于被所述计算机执行以实现权利要求1-7任一项所述的非严格对齐的小样本不同源图像转换方法。

技术总结
本发明属于图像转换技术领域,具体涉及了一种非严格对齐的小样本不同源图像转换方法、系统及设备,旨在解决现有深度学习的模型训练样本图像少、获取困难,导致训练好的模型性能较低的问题。本发明包括:构建图像转换模型,其生成器基于编码器、转换器和解码器构建,判别器基于卷积神经网络和激活函数构建;获取X类别和Y类别图像对,并进行图像增强;通过增强后的图像对进行图像转换模型的训练;通过训练好的图像转换模型,将获取的一个或少量X类别图像转换为Y类别图像。本发明通过非严格对齐方法构建容易获得的图像类别向不容易获得的图像的转换,有效提升不容易获得的图像的目标特征学习效果。征学习效果。征学习效果。


技术研发人员:杨明浩 张晓凯
受保护的技术使用者:中国科学院自动化研究所
技术研发日:2022.07.21
技术公布日:2022/9/20
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1