HiSD 如何设置custom datasets?

如何设置custom datasets?

Open Ziyan-swu opened this issue 2 years ago • 6 comments

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

Jul 20 '22 07:07 Ziyan-swu

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

Jul 20 '22 13:07 imlixinyang

感谢您的回答！请问train set是指那个文件呢？截屏2022-07-21 11 52 40

Jul 21 '22 03:07 Ziyan-swu

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

Jul 21 '22 05:07 imlixinyang

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

Jul 21 '22 05:07 Ziyan-swu

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

Jul 21 '22 05:07 imlixinyang

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

Jul 21 '22 06:07 Ziyan-swu