Error IndexError: boolean index did not match indexed array along dimension 0; dimension is 780 but corresponding boolean dimension is 799

[amitkayal]

Hi,

Getting the following error. Is this due to size mismatch?

Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to /root/.cache/torch/hub/checkpoints/resnet18-5c106cde.pth
100% 44.7M/44.7M [00:00<00:00, 307MB/s]
Traceback (most recent call last):
File "makeup.py", line 86, in
image = hair(image, parsing, part, color)
File "makeup.py", line 54, in hair
changed[parsing != part] = image[parsing != part]
IndexError: boolean index did not match indexed array along dimension 0; dimension is 780 but corresponding boolean dimension is 799

[amitkayal]

Here is the input file

[yaroslavmavliutov]

Hi @amitkayal

You can add np.repeat(parsing[:, :, np.newaxis], 3, axis=2) to hair function. It works

[qihuijia]

Hi @amitkayal

You can add np.repeat(parsing[:, :, np.newaxis], 3, axis=2) to hair function. It works

hi, where should I add this code?