CN116403026A - 一种基于多类别风格提取的风格迁移方法 - Google Patents
一种基于多类别风格提取的风格迁移方法 Download PDFInfo
- Publication number
- CN116403026A CN116403026A CN202310229019.2A CN202310229019A CN116403026A CN 116403026 A CN116403026 A CN 116403026A CN 202310229019 A CN202310229019 A CN 202310229019A CN 116403026 A CN116403026 A CN 116403026A
- Authority
- CN
- China
- Prior art keywords
- style
- adaptive
- image
- encoder
- category
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 230000005012 migration Effects 0.000 title claims abstract description 28
- 238000013508 migration Methods 0.000 title claims abstract description 28
- 238000000034 method Methods 0.000 title claims abstract description 22
- 238000000605 extraction Methods 0.000 title claims abstract description 20
- 238000010606 normalization Methods 0.000 claims abstract description 12
- 238000012549 training Methods 0.000 claims abstract description 11
- 230000008569 process Effects 0.000 claims abstract description 6
- 238000013528 artificial neural network Methods 0.000 claims abstract 3
- 238000011176 pooling Methods 0.000 claims description 12
- 230000003044 adaptive effect Effects 0.000 claims description 11
- 230000003595 spectral effect Effects 0.000 claims description 7
- 230000004913 activation Effects 0.000 claims description 6
- 238000006243 chemical reaction Methods 0.000 claims description 6
- 230000006870 function Effects 0.000 claims description 3
- 230000006872 improvement Effects 0.000 claims description 2
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 claims 2
- 238000013256 Gubra-Amylin NASH model Methods 0.000 claims 1
- 230000000694 effects Effects 0.000 abstract description 6
- 238000001228 spectrum Methods 0.000 abstract description 5
- 239000000284 extract Substances 0.000 abstract 2
- 230000004927 fusion Effects 0.000 abstract 1
- 230000007246 mechanism Effects 0.000 description 5
- 238000010586 diagram Methods 0.000 description 4
- 239000011159 matrix material Substances 0.000 description 4
- 238000013519 translation Methods 0.000 description 4
- 125000004122 cyclic group Chemical group 0.000 description 2
- 238000013507 mapping Methods 0.000 description 2
- 230000000007 visual effect Effects 0.000 description 2
- 230000006978 adaptation Effects 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000005457 optimization Methods 0.000 description 1
- 230000010355 oscillation Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N20/00—Machine learning
- G06N20/20—Ensemble learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Evolutionary Computation (AREA)
- Software Systems (AREA)
- Medical Informatics (AREA)
- Artificial Intelligence (AREA)
- Computing Systems (AREA)
- General Physics & Mathematics (AREA)
- Physics & Mathematics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Health & Medical Sciences (AREA)
- Databases & Information Systems (AREA)
- General Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Data Mining & Analysis (AREA)
- General Engineering & Computer Science (AREA)
- Mathematical Physics (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于多类别风格提取的风格迁移方法,包括,内容编码器提取源数据域的内容特征;多类别风格提取编码器提取目的数据域不同类别的风格特征,并将多类别的风格特征进行统计;解码器针对内容特征和多类别风格特征进行特征融合,以此生成与目的数据域相似的样本;并利用对比学习方法使神经网络在风格迁移过程中,不会对源数据域的内容特征进行改变。并在判别器中添加谱归一化操作,以此减少网络的训练。本发明方法能够有效的使用非配对图像输入到神经网络进行训练,以此得到较佳的风格迁移效果。
Description
技术领域
本发明涉及风格迁移领域,尤其涉及一种基于多类别风格提取的风格迁移方法。
背景技术
Image-to-Image Translation(I2IT)是一类计算机视觉任务,其旨在将输入图像转换为输出图像。这种任务可以应用于许多实际场景中,例如图像修复、图像翻译、图像风格转换等。在转换过程中,输出结果应保持源域图像的内容特征不变,同时使网络能够学习目标域的风格特征,这是I2IT的重要目标。
Pix2Pix是一个基于生成对抗网络(GAN)的有监督图像到图像转换模型,最早由Isola等人在2017年提出。Pix2Pix的生成器采用了U-Net的结构,判别器采用了PatchGAN的结构,判别器会对图像的每个小区域进行判别,这可以提高判别器的效率和精度。但Pix2Pix模型对于输入图像的质量非常敏感,如果输入图像存在噪声、模糊或失真等问题,模型可能会产生不准确的输出。
MUNIT(Multimodal Unsupervised Image-to-image Translation)是一种用于图像转换的无监督模型,它的主要思想是将图像的内容和风格分开处理,并利用对抗损失(Adversarial Loss)和循环一致性损失函数(Cycle-consistency Loss)来训练模型。MUNIT包括三个主要模块:内容编码器网络、样式编码器和解码器网络。
在训练过程中,MUNIT使用对抗损失来优化编码器和解码器网络,使其能够学习到如何从随机噪声中生成具有特定视觉风格的图像。同时,利用循环一致性损失函数来保留原始图像的内容信息,并在不同的视觉风格之间进行转换。但是该模型在训练时需要消耗较多的计算资源和内存资源。
Taesung Park等人提出CUT模型,与其他无配对图像翻译模型相比,CUT的目标是学习两个图像之间的块映射,而不是像素到像素的映射。其主要由两个主要组件组成:生成器和判别器。生成器从一个域生成图像,并通过对比学习将其转换为另一个域。具体来说,生成器包括两个子网络:编码器和解码器。编码器将输入图像编码为一组特征,解码器将这些特征转换为另一个域中的图像。对比学习使用了一个判别器网络,它被训练来区分翻译图像和真实图像。但这可能会导致生成的图像过于相似,缺乏真实样本中的多样性和复杂性。
综上所述,I2IT任务虽然得到的广泛的研究,但是现存的I2IT方法还存在迁移效果不佳等问题。因此,该发明提出一种基于多类别风格提取方法并将其应用到I2IT任务中,以此提高风格迁移效果。
发明内容
本发明目的是为了解决风格迁移过程中,迁移效果不佳问题。提出了一种基于多类别风格提取的风格迁移方法,以此提升图像的迁移效果。
为实现上述目的,本发明提供如下技术方案:提出了一种基于多类别风格提取的风格迁移方法,包括如下步骤:
1)从风格迁移官方数据集中加载源域A和目标域B的图像数据;
2)将源域图像A输入到内容编码器(Content_Encoder)中,得到关于源域图像A的内容编码Encode_A;
3)将目标域图像B输入到风格编码器(Style_Encoder)中,得到关于目标域图像的风格编码style_B;
4)将源域图像A的内容编码Encode_A和关于目标域的风格编码style_B输入到解码器(Decoder)中,得到生成的目标图像Fake_B;
5)将Fake_B图像从计算图上分离,传入带有Spectral Normalization的判别器中,输出得到判别结果,将结果结合对抗损失得到生成图像的损失值;
6)将目标域的真实图像B传入带有Spectral Normalization的判别器中,输出得到判别结果,将结果结合对抗损失得到真实图像的损失值;
7)将生成图像的损失值和真实图像的损失值相加得到判别器的损失值;
8)反向传播判别器的损失值,更新判别器的梯度;
9)将生成图像Fake_B和真实图像A传入Content_Encoder得到对比损失PatchNCEloss;
10)将判别器的判别结果和PatchNCE loss相加得到生成器损失值;
11)将判别器和生成器的损失值相加得到总损失值;
12)设置判别器的梯度为不可以更新状态,利用生成器的损失值计算得到生成器的梯度更新值,反向传播生成器的梯度,更新生成器;
13)通过不断地迭代训练得到最终的模型。
所述步骤5)中,得到的对抗损失,具体如下:
式(1)中G代表域A→B的生成器,D代表判别器;
式(1)中a~A,b~B分别代表A域和B域中的训练样本;
上述步骤10)中,得到的对比损失PatchNCE loss,具体计算如下:
式(2)中v代表生成图像Fake_B中将要进行查询的样本点,v+代表源域图像A中对应位置的样本点,v-代表源域图像A中其它位置的样本点,τ是超参数,代表温度系数,将它设置为0.07;
所述步骤11)中,得到的总损失,具体计算如下:
L(G,D,A,B)=LGAN(G,D,A,B)+λ*PatchNCE (3)
式(4)中,λ代表的是编码的相关性;
所述步骤15)中,迭代训练,具体优化目标为:
式(4)中,G*和D*为最优情况下生成器和判别器;
本发明的有益效果是:
一、本发明基于风格迁移问题,提出一种基于多类别风格提取的风格迁移方法,该方法能够有效提升风格迁移效果。
二、本发明采用一种注意力机制,对输入图像的重要特征进行增强。
三、本发明采用一种全新的多种风格提取方法,对迁移结果进行提升。
四、本发明采用了谱归一化操作对判别器网络进行约束,从而缓解网络的振荡。
附图说明
图1是本发明网络模型图;
图2是本发明实例多样式提取的生成器网络模型图;
图3是本发明实例注意力机制模型图;
图4是本发明实例判别器网络模型图。
具体实施方式
为了使本发明的目的及技术方案更加明显,以下结合附图及实施例,对本发明进行进一步详细说明。应当理解,此处所描述的具体实施仅仅用于解释本发明,但不用于限定本发明。此外,下面所描述的本发明各个实施方式中涉及到的技术特征只要彼此之间未构成冲突就可以相互结合。
本发明提供一种基于多类别风格提取的风格迁移方法,网络模型的生成器采用多类别风格提取的Style Encoder和带注意力机制的Content Encoder,本网络模型的判别器使用了谱归一化操作,具体步骤如下:
步骤S1,使用face2anime数据集,其中包括源域图像和目标域图像,并将图像输入网络模型中,网络模型如图1所示。
步骤S2,如图2所示,对Style_Encoder进行改进,提出了一种新的提取多种风格的编码器Mutli-Style Extraction Encoder,目标域图像在经过多次卷积层之后得到多类别的目标域风格编码,源域图像经过内容编码器得到内容编码。
步骤S2中,提取多种风格的编码器的具体步骤为:
S21,给定一个特征图作为输入张量:
Finput=(fijk)H×W×c s.t.i=1,2,,,H;j=1,2,,,W;k=1,2,,,C;
其中Finput是指输入张量,(H,W,C)指输入张量的形状,fijk代表Finput中位置为(i,j,k)的值。
S22,将Finput输入到卷积层(Conv1ijnm)7*7*C*64,得到卷积之后的结果(Conv1_outijk)H*W*64;
Conv1_outijk指第k个卷积核卷积fijk之后的结果。
S22,将(Conv1_outijk)H*W*64进行ReLu激活,得到(ReLu1_outijk)H*W*64;
Conv2_outijk指第k个卷积核卷积ReLU1_outijk之后的结果。
Conv3_outijk指第k个卷积核卷积ReLU2_outijk之后的结果。
Conv4_outijk指第k个卷积核卷积ReLU3_outijk之后的结果。
Conv5_outijk指第k个卷积核卷积ReLU4_outijk之后的结果。
S214,将AdaAvgPooling_out1*1*256、AdaMaxPooling_out1*1*256和AdaStdPooling_out1*1*256在通道维度上进行连接得到多种类风格矩阵(Cat_outijk)1*1*768。
S215,将(Cat_outijk)1*1*768输入到卷积层(Conv6ijnm)1*1*768*8,得到卷积之后的结果(MultiStyle_featureijk)1*1*8;
MultiStyle_featureijk指第k个卷积核卷积Cat_outijk之后的结果。
S216,将提取到的多种类风格(MultiStyle_featureijk)1*1*8注入到编码器中。
步骤S3,将风格编码注入到注意力机制的解码器中,以此实现目标域风格特征和源域内容特征的结合,注意力机制结构如图3所示。
步骤S4,如图4所示,对判别器网络进行改进,通过增加Spectral Normalization,以此约束原始判别器的卷积操作,从而稳定网络的训练过程。
步骤S4中,Spectral Normalization的具体操作为:
S41,给定卷积操作的权重矩阵为Wn*m;
S42,初始化一个随机的向量v1*m;
步骤S5,对S2和S3中改进的模型进行训练,得到一个全新的风格迁移模型MSE-GAN。
本领域的技术人员容易理解,以上所述仅为本发明的较佳实施例而已,并不用以限制本发明,凡在本发明的精神和原则之内做任何修改,等同替换和改进等,均应包含在本发明的保护范围之内。
Claims (4)
1.一种基于多类别风格提取的图像风格迁移方法,其特征在于,包括如下步骤:
采用公开的风格迁移数据集训练MSE-GAN,训练集中样本为待风格迁移的源域图像和有目标风格的目的域图像,将训练样本输入到神经网络中,反复迭代,以此得到训练好的风格迁移网络;将测试集的源域图像输入到训练好的风格迁移网络中,得到与目的域相似的全新图像,在转换过程中,源域图像的几何特征并不会发生改变;
所述MSE-GAN包含Multi-Style Extraction Encoder、Content Encoder、Decoder和带有Spectral Normalization的Discriminator;Multi-Style Extraction Encoder包括ConvBlock、自适应平均池化AdaptiveAvgPooling、自适应最大池化AdaptiveMaxPooling和自适应方差池化AdaptiveStdPooling,其中ConvBlock表示卷积和实例归一化操作,激活函数是Relu;Content Encoder包括Downsample、Statistical Attention和Residual Block。Decoder包括AdaIN Residual Block、Statistical Attention和Upsample;SN-Discriminator包含3个带有Spectral Normalization的ConvBlock。
2.如权利要求1所述方法,其特征在于,多类别风格提取编码器Multi-StyleExtraction Encoder的结构包括:
大小为H*W*C的输入特征图在通道维度上分别进行自适应平均池化、自适应最大池化和自适应方差池化,得到三个大小为1*1*C的特征图矩阵,拼接三个特征图矩阵输入到卷积层,卷积层中卷积核的个数为C,卷积层最终输出得到多类别的风格特征矩阵。
3.如权利要求1所述,其特征在于,新的风格迁移模型MSE-GAN的改进包括:
将Multi-Style Extraction Encoder与Content Encoder、Decoder和带有SpectralNormalization的Discriminator进行结合,得到新的风格迁移模型MSE-GAN。
4.如权利要求1所述,其特征在于,训练MSE-GAN模型:
根据MSE-GAN的结构,结合对比学习方法,以此保证图像几何特征的一致性,得到新的风格迁移模型MSE-GAN。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202310229019.2A CN116403026A (zh) | 2023-03-10 | 2023-03-10 | 一种基于多类别风格提取的风格迁移方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202310229019.2A CN116403026A (zh) | 2023-03-10 | 2023-03-10 | 一种基于多类别风格提取的风格迁移方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN116403026A true CN116403026A (zh) | 2023-07-07 |
Family
ID=87009388
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202310229019.2A Pending CN116403026A (zh) | 2023-03-10 | 2023-03-10 | 一种基于多类别风格提取的风格迁移方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN116403026A (zh) |
-
2023
- 2023-03-10 CN CN202310229019.2A patent/CN116403026A/zh active Pending
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN110189334B (zh) | 基于注意力机制的残差型全卷积神经网络的医学图像分割方法 | |
CN111079532B (zh) | 一种基于文本自编码器的视频内容描述方法 | |
CN107798381B (zh) | 一种基于卷积神经网络的图像识别方法 | |
CN109711426B (zh) | 一种基于gan和迁移学习的病理图片分类装置及方法 | |
CN111292330A (zh) | 基于编解码器的图像语义分割方法及装置 | |
CN110570433B (zh) | 基于生成对抗网络的图像语义分割模型构建方法和装置 | |
CN110796619A (zh) | 一种图像处理模型训练方法、装置、电子设备及存储介质 | |
CN112418292B (zh) | 一种图像质量评价的方法、装置、计算机设备及存储介质 | |
CN111582483A (zh) | 基于空间和通道联合注意力机制的无监督学习光流估计方法 | |
CN111899168B (zh) | 一种基于特征增强的遥感影像超分辨率重建方法和系统 | |
CN114511576B (zh) | 尺度自适应特征增强深度神经网络的图像分割方法与系统 | |
CN112884758B (zh) | 一种基于风格迁移方法的缺陷绝缘子样本生成方法及系统 | |
CN114418030A (zh) | 图像分类方法、图像分类模型的训练方法及装置 | |
CN111931779A (zh) | 一种基于条件可预测参数的图像信息提取与生成方法 | |
CN113139446A (zh) | 一种端到端自动驾驶行为决策方法、系统及终端设备 | |
CN113763268A (zh) | 人脸图像盲修复方法及系统 | |
CN112149526A (zh) | 一种基于长距离信息融合的车道线检测方法及系统 | |
CN115358952A (zh) | 一种基于元学习的图像增强方法、系统、设备和存储介质 | |
CN114494387A (zh) | 一种生成数据集网络模型及雾图生成方法 | |
CN107729885B (zh) | 一种基于多重残差学习的人脸增强方法 | |
CN116306780B (zh) | 一种动态图链接生成方法 | |
CN112560719A (zh) | 基于多尺度卷积-多核池化的高分辨率影像水体提取方法 | |
CN112766099A (zh) | 一种从局部到全局上下文信息提取的高光谱影像分类方法 | |
CN116977694A (zh) | 一种基于不变特征提取的高光谱对抗样本防御方法 | |
CN115965836A (zh) | 一种语义可控的人体行为姿态视频数据扩增系统及方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |