test with custom image：image shape

minar09 / cp-vton-plus

Official implementation for "CP-VTON+: Clothing Shape and Texture Preserving Image-Based Virtual Try-On", CVPRW 2020

https://minar09.github.io/cpvtonplus/

MIT License

352 stars 121 forks source link

test with custom image：image shape #63

Closed mcc1999 closed 3 years ago

mcc1999 commented 3 years ago

I download the author's model, and want to test with my cunstom image. First, I run the GMM model to get wrap-cloth, but I get the error as follow: The shape of im is (3, 256,192), and the pcm shape is (256,192, 3). When im * pcm, the error happened. After reshaping the pcm to (3,256,192), I get another error as follow: I don't know where the wrong located, it's wrong with im shape? Or pcm shape? Or my custom image is in a wrong shape? Should I reshape the im shape? @minar09

minar09 commented 3 years ago

Hi @mcc1999 , sorry for the inconvenience. Please see this comment for the error details: https://github.com/minar09/cp-vton-plus/issues/60#issuecomment-817379874