search

imlixinyang / HiSD

Code for "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).
Other
391 stars 49 forks source link

如何设置custom datasets? #37

Closed Ziyan-swu closed 2 years ago

Ziyan-swu commented 2 years ago

作者您好，想问一下如何设置custom datasets?需要自己在网上搜索TAGs的相关图片吗？还是使用提供的数据集呢？我在CelebA-HQ里没有找到符合条件的数据集

截屏2022-07-20 15 23 06
imlixinyang commented 2 years ago

custom datasets一般是用在普通的以文件夹名区分类别的数据集，对于Celeba来说直接使用celeba对应的预处理脚本就好～具体的脚本文件是proprecessors/celeba-hq.py，按照README指定好图像文件夹和label文件路径即可。

Ziyan-swu commented 2 years ago

感谢您的回答！请问train set是指那个文件呢？

截屏2022-07-21 11 52 40
imlixinyang commented 2 years ago

celeba的话应该是根据这个来：

image
Ziyan-swu commented 2 years ago

好的，您的意思是根据上面的命令行预处理之后，就不需要使用我提到train-set和custom dataset（即截图中提到的部分）进行设置了吗？

截屏2022-07-21 13 40 10
imlixinyang commented 2 years ago

是的，得到txt文件之后就可以运行训练脚本了～

Ziyan-swu commented 2 years ago

好的～，我懂您的意思了，谢谢！