如何设置custom datasets?

[AriaZiyanYang]

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

[imlixinyang]

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

[AriaZiyanYang]

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

[imlixinyang]

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

[AriaZiyanYang]

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

[imlixinyang]

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

[AriaZiyanYang]

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