Error during learning process about dualstylegan HOT 5 CLOSED

I changed it to jpg because I thought the format of the picture might be a problem, but there must be a problem.

So the error occurs when you use jpg or png?
The code uses mean and std to adjust 3 channels of your data.

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

It seems that your data has 4 channels.

    transform2 = transforms.Compose(
    [
        transforms.ToTensor(),
        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5), inplace=True),
    ])     

File "finetune_dualstylegan.py", line 529, in
Simg = transform2(Simg).unsqueeze(dim=3)

My original code is Simg = transform2(Simg).unsqueeze(dim=0)
Did you make modifications to the code, which leads to the error?

from dualstylegan.

@williamyang1991 It was originally a png format, but I changed it to jpg, but there is still an error. I tried your original code, but it's the same! Do you think the png format is a problem?

from dualstylegan.

you can add
print(np.array(Simg).shape) before
Simg = transform2(Simg).unsqueeze(dim=0)
to see the shape of your Simg.
It should be (1024, 1024, 3)
otherwise, there is something wrong with your style image.

from dualstylegan.

Alpha Channel was the problem! Thanks for letting me know. One question is that input image is definitely channel 3, but Simg is still channel 4. Can I assume there was a malfunction in the middle?

from dualstylegan.

My code is just read the image and transform it.

Simg = Image.open(os.path.join(path, filename))
Simg = transform2(Simg).unsqueeze(dim=0)

It was originally a png format, but I changed it to jpg

It should be that you incorrectly changed png to jpg without discarding its alpah channel.

from dualstylegan.