文件名称:SieveNet:这是SieveNet的非官方实现
文件大小:839KB
文件格式:ZIP
更新时间:2024-05-24 12:57:37
pytorch pytorch-implementation vton 2d-virtual-try-on vitual-tryon
筛网 这是“ SieveNet:用于基于映像的鲁棒虚拟尝试的统一框架”的非官方实现。 纸可以在找到 数据集下载与处理 数据集下载说明和数据集链接可从和官方找到将数据集放入data文件夹 用法 通过pip install -r requirements.txt复制回购和安装pip install -r requirements.txt 训练 粗到细变形模块 在config.py设置self.datamode='train'和self.stage='GMM' 然后运行python train.py 您可以在tensorboard中进行转换时观察结果,如下所示 条件分割蒙版生成模块 在config.py设置self.datamode='Train'和self.stage='SEG' 然后运行python train.py 分割辅助纹理翻译模块 在config.py设置self.datamode
【文件预览】:
SieveNet-master
----.gitignore(8B)
----images()
--------test_tom.png(229KB)
--------train_gmm.png(165KB)
--------test_gmm.jpg(118KB)
--------train_tom.png(231KB)
--------train_seg.jpeg(12KB)
----predict_pose.py(4KB)
----visualization.py(3KB)
----requirements.txt(134B)
----data()
--------grid.png(802B)
--------train_pairs.txt(361KB)
--------test_pairs.txt(52KB)
----LICENSE(1KB)
----inference.py(7KB)
----README.md(4KB)
----unet.py(4KB)
----config.py(2KB)
----test.py(6KB)
----dataset.py(9KB)
----pose()
--------pose_deploy_linevec.prototxt(45KB)
----gmm.py(13KB)
----losses.py(5KB)
----train.py(12KB)