search

imlixinyang / HiSD

Official pytorch implementation of paper "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).
Other
390 stars 49 forks source link

作者您好，请问下有关数据集的问题 #24

Open zhangqian001 opened 3 years ago

zhangqian001 commented 3 years ago

我按照您的提示下载了CelebAmask-HQ数据集，但是我不知道img_path, label_path, target_path之类的具体怎么设置，您可以给我一个示例吗？我是初学者。此外，Label_path这个我也不太会，我下载的文件里没有label标签文件，难道是mapping.txt？但是我设置了这个之后，提示我没有这个路径。谢谢作者大大。

imlixinyang commented 3 years ago

img_path指的是原始图片的文件夹位置，label_path指的是标注文件位置，target_path指的是处理后的文件存放的文件夹位置，一般来说你整理好下载的数据集后，可以很方便地得到需要的目标文件。这一步的目的是避免重复地去存放图像文件和标注。 按照celeba数据集为例，img和label分别为Align&Cropped Images和Attributes Annotations。 image 在训练的时候，一般情况下，只需要在config文件，也就是configs/celeba-hq.yaml指定你的target文件就可以了。 这些都是数据集本身自带的标注，不需要你再进行分类，CelebAmask-HQ中应该有annotations文件，方便的话可以截图给我看看下载下来的压缩包里的文件结构。