switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About how to fix the BUG of g_partition.py

xilinlan opened this issue · comments

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.