【文件属性】:
文件名称:DomainIntersectionDifference:Pytorch实现的“域交集和域差异”(ICCV 2019)
文件大小:893KB
文件格式:ZIP
更新时间:2021-03-15 00:25:48
deep-learning generative-model image-to-image disentangled-representations Python
域交集和域差异( )。
Pytorch实现“域交集和域差异”(ICCV 2019)
先决条件
Python 2.7 / 3.6
火炬0.4
下载并准备数据
下载celeba数据集。 创建一个celeb目录,并将img_align_celeba文件夹和list_attr_celeba.txt放在其中。
您可以使用提供的脚本preprocess.py将celebA拆分为上述格式(根据您选择的属性使用A和B)。 例如,您可以使用以下命令运行脚本:
python preprocess.py --root ./celeba/img_align_celeba --attributes ./celeba/list_attr_celeba.txt --dest ./smile_glasses
您还可以使用自己的自定义数据集,只要它遵循以下格式:
root/
trainA/
【文件预览】:
DomainIntersectionDifference-master
----models()
--------models.py(10KB)
----eval.py(2KB)
----train()
--------train.py(8KB)
----utils.py(20KB)
----preprocess.py(6KB)
----DiagramsDomainIntersection.png(896KB)
----README.md(2KB)