【源码】基于Python的虚拟换衣系统

项目简介

本项目是基于Python构建的虚拟换衣系统，借助深度学习与图像处理技术达成虚拟试衣功能。它以VITON - HD论文为基础，采用预训练模型，用户只需输入一张带有背景的人物照片和一张衣服照片，系统就能自动完成衣服、人体分割和姿态捕捉等过程，生成换装后的图像。

项目的主要特性和功能

1. 简化输入：用户仅需提供一张有背景的人物图像和一张衣服图像，即可实现虚拟换衣，降低了对输入数据的要求。
2. 多模块处理：包含衣服分割、人体分割和姿态检测模块，能精准提取所需信息。
3. 高分辨率支持：可处理高分辨率图像，在1024x768尺度上也能获得较好的换衣效果，且能有效消除伪影。
4. 细节复现：能够较好地复现衣服的图案、款式、花纹，同时清晰保持原模特的脸部、手部、裤子等部位。

安装使用步骤

环境准备

VITON - HD

• Windows运行环境：Windows 10系统，配备RTX 3070 GPU，cudnn版本为7.6.5，Cuda版本为3.7.10。
• Anaconda虚拟环境：Python 3.8.18，cudatoolkit 8.0，torch 1.13.0+cu117，torchaudio 0.13.0+cu117，torchvision 0.14.0+cu117。

各模块依赖

• 衣服分割：使用U2 - Net预训练好的模型，无需额外安装库函数。

运行步骤

1. 已有批处理命令run.bat，在不同终端运行时，将其中的绝对路径改为本机上的路径。
2. 在命令行输入run person_00 cloth.jpg（person是人物图像名称，不包括后缀，cloth.jpg是衣服图片名称）。
3. 运行后会自动修改datasets\test_pairs.txt中的人物 - 服装对，换装结果保存为results\test\person_cloth.jpg。

注意事项

使用前要将传入的两张图片修改为768*1024的大小，否则会因图像尺寸不匹配导致报错。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】