face-parsing.PyTorch icon indicating copy to clipboard operation
face-parsing.PyTorch copied to clipboard

dataset I think there may be a bug

Open AN-ZE opened this issue 1 year ago • 0 comments

In the code, the image is resized to 512, but the mask does not perform any operations. Their shapes are different https://github.com/zllrunning/face-parsing.PyTorch/blob/d2e684cf1588b46145635e8fe7bcc29544e5537e/face_dataset.py#L47 https://github.com/zllrunning/face-parsing.PyTorch/blob/d2e684cf1588b46145635e8fe7bcc29544e5537e/face_dataset.py#L48

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

AN-ZE avatar May 30 '23 03:05 AN-ZE