search

eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework
https://eladrich.github.io/pixel2style2pixel/
MIT License
3.19k stars 570 forks source link

Could I get the "sketch-face" dataset you mentioned in your article? #236

Closed gzhhhere closed 2 years ago

gzhhhere commented 2 years ago

Hi! Thank you for your excellent work. Could you please share the "sketch-face" data set with me, in whatever way is convenient to you?

yuval-alaluf commented 2 years ago

Unfortunately, I am unable to share the sketch dataset due to licensing. However, you can use the following script for recreating the data: https://github.com/eladrich/pixel2style2pixel/blob/master/scripts/generate_sketch_data.py

gzhhhere commented 2 years ago

Thanks for your quick reply, it will save me a lot of time. I didn't even notice the script existed.

gzhhhere commented 2 years ago

I can run the script on CPU slowly , but when i try it on GPU, it just stuck at line 59 'pred = model.cuda().forward(data.cuda()).float()'. Have you ever had a similar problem?

yuval-alaluf commented 2 years ago

I don't remember that problem. Did you make sure you have the torch version specified in the script?

gzhhhere commented 2 years ago

yes, i am sure. When i wait it run on GPU, i get "runtimeerror: cublas runtime error : the gpu program failed to execute at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/thc/thcblas.cu:249" ... have no idea about it,

gzhhhere commented 2 years ago

i solve the problem by choice a v100 server, i was using a 3090 before the time.