dataset I think there may be a bug

[AN-ZE]

In the code, the image is resized to 512, but the mask does not perform any operations. Their shapes are different

face-parsing.PyTorch/face_dataset.py

Line 47 in d2e684c

img = img.resize((512, 512), Image.BILINEAR)

face-parsing.PyTorch/face_dataset.py

Line 48 in d2e684c

label = Image.open(osp.join(self.rootpth, 'mask', impth[:-3]+'png')).convert('P')

The latter is changed through RandomScale.
I think add a line of code
label = label.resize((512, 512), Image.NEAREST)