wty-ustc / HairCLIP

[CVPR 2022] HairCLIP: Design Your Hair by Text and Reference Image
GNU Lesser General Public License v2.1
524 stars 68 forks source link

Can I use my own image test? #12

Closed liuzhuangyuan closed 2 years ago

liuzhuangyuan commented 2 years ago

Hello, can I use my own image for the resend test, I found that the input was test_face.pt (test data set ?) file, and I did not find the input image content in the code, The only thing that feels like an input image is w(w=torch.Size([1, 18, 512])), But it's not the size of a picture

wty-ustc commented 2 years ago

Sure, but you need to use e4e to get the latent code of the image.

liuzhuangyuan commented 2 years ago

Ok, thanks for your reply. I'll try it