switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.
2.04k stars 343 forks source link

About how to fix the BUG of g_partition.py #118

Open xilinlan opened 5 months ago

xilinlan commented 5 months ago

1、 line 8 s_label = 'CelebAMask-HQ-label' this s_label in g_mask.py generate 2、 line 33 image_list = pd.read_csv('CelebA-HQ-to-CelebA-mapping.txt', delim_whitespace=True, header=None) the mapping.txt has title header, so just use the header default parameter is ok the correct use: image_list = pd.read_csv('CelebA-HQ-to-CelebA-mapping.txt', delim_whitespace=True) 3、 line 38 for idx, x in enumerate(image_list.loc[:, 1]): it has a error. The proper way is for idx, x in enumerate(image_list.loc[:1]): but it not end image_list.loc[:1] just has two line, and the content is incorrect the right way to use is as below: for idx, x in enumerate(image_list.loc[:, 'orig_idx']):

The above is the problem I encountered during the usage and the solution. If there are any errors, please kindly forgive me.