首页|基于去噪扩散生成模型的虚拟试衣方法

基于去噪扩散生成模型的虚拟试衣方法

扫码查看
采用基于蒙板合成策略的虚拟试衣,容易产生严重遮挡,同时由于服装特征提取不足,会使衣物的复杂细节发生改变.对此,使用扩散模型作为虚拟试衣任务的主干框架,提出基于去噪扩散生成模型的虚拟试衣方法.在每个时间点根据高斯分布对模特图像加入噪声,逐渐将高斯噪声分布转换为生成模型所训练的数据分布.将目标衣物图像作为条件来引导扩散模型反向采样,生成模特试衣之后的干净背景图像.使用U-net结构作为噪声预测网络的主干网络,将基于交叉注意力机制的纹理对齐模块嵌入U-net结构的不同层中.为了保证模特身份信息和背景图像不变,采用混合扩散方法,通过掩码对前后两阶段相同步骤的噪声图像进行混合,使原本多样性扩散生成结果转变为仅限于换衣区域的图像重建.试验结果表明,采用基于去噪扩散生成模型的虚拟试衣方法,生成图像的峰值信噪比和结构相似性指数都有所提高,平均FID指标相对减小.
The virtual fitting based on mask synthesis strategy is easy to cause serious occlusion,and complex detail of the clothing will be changed due to the insufficient clothing feature extraction.In this regard,diffusion model was used as the backbone framework of virtual fitting task,and a virtual fitting method based on denoising diffusion generation model was proposed.At each time point,noise is added to the model image according to Gaussian distribution,and Gaussian noise distribution is gradually transformed into the data distribution for which the generation model is trained.The target clothing image is used as a condition to guide diffusion model to inverse sample to generate a clean background image after the model is tried on.The U-net structure is employed as the backbone network of the noise prediction network,and the texture alignment module based on cross-attention mechanism is embedded in different layers of the U-net structure.In order to ensure that the identity information of the model and the background image are unchanged,the mixed diffusion method is used to mix the noise images of the same step in the two stages through the mask,so that the original diversity diffusion generation result is transformed into the image reconstruction limited to the changing area.The experimental result shows that the peak signal-to-noise ratio and structural similarity index of the generated image are improved,and the average FID index is relatively reduced by using the virtual fitting method based on denoising diffusion generation model.

Diffusion ModelVirtualFitting

钱丽丹、黄明、臧福星、黄小杰、蒋鑫

展开 >

江苏理工学院机械工程学院 江苏常州 213001

扩散模型 虚拟 试衣

2024

装备机械
上海电气(集团)总公司

装备机械

影响因子:0.158
ISSN:1662-0555
年,卷(期):2024.(3)