HiSD icon indicating copy to clipboard operation
HiSD copied to clipboard

更换数据集

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

作者您好,我想对HiSD更换数据集进行训练,对于CelebA数据集您通过label.txt文件对数据进行预处理,里面有关于刘海、眼镜等属性的信息,请问关于HiSD是否有已经标注好的其他数据集呢~

Ziyan-swu avatar Aug 19 '22 04:08 Ziyan-swu

每个数据集的标注文件也有很多不一样,比如FFHQ-aging里是一个csv文件,RaFD里是在文件名中,这些不一致导致了我们采用先行预处理这些标注的脚本来让HiSD config接收到的始终是一个以每个tag的attribute为独立文件,以文件路径+每个tag的无关标注这样的形式。 因此,需要更改数据集的话只需要自己编写相应的脚本来将这些标注信息转换到与celeba示例中一样,并且修改你的config文件即可~

imlixinyang avatar Aug 19 '22 05:08 imlixinyang

谢谢!

Ziyan-swu avatar Aug 19 '22 05:08 Ziyan-swu