imlixinyang / HiSD

Official pytorch implementation of paper "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

作者您好,请问下有关数据集的问题

zhangqian001 opened this issue · comments

commented

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

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