Error : The size of tensor a (1050) must match the size of tensor b (500)

Taeu / Style-Your-Hair

Official Pytorch implementation of "Style Your Hair: Latent Optimization for Pose-Invariant Hairstyle Transfer via Local-Style-Aware Hair Alignment (ECCV 2022)"

295 stars 40 forks source link

Error : The size of tensor a (1050) must match the size of tensor b (500) #21

Open girithodu opened 10 months ago

girithodu commented 10 months ago

How do I get past this issue if the source image tensor size is different from the target picture?

dev-kodexo commented 9 months ago

I am getting same error tensor size of a should be equal to tensor size of b

Fahad-Aslam commented 9 months ago

Try to convert both the source and target images, ensuring that they share identical dimensions (e.g., 1024 x 1024) and equivalent color depths (e.g., 24 bits per pixel).

kampelmuehler commented 6 months ago

Inputs apparently must be 1024x1024 for it to work.