ermongroup/SDEdit

CelebA-HQ pth missing

ucalyptus2 opened this issue · 7 comments

aferriss commented

+1 I'd also love to try out the celeba model.

chenlin9 commented

Thank you for the question.

You can directly load the CelebA-HQ model by using "—config celeba.yml". However, you will need to create your own mask and image, upload the file to the "colab_demo directory", and then change "—npy_name" argument to be the name of the file. More details can be found at
https://github.com/ermongroup/SDEdit/blob/main/runners/image_editing.py#L108
We are currently working on an interactive interface for creating masks and editing images.

Let me know if you have any other questions. Thank you.

sorayutmild commented

@chenlin9 How can we create our own mask and image. Is it .pth file?

sorayutmild commented

I've already solved it. just using torch.save()
for example.
torch.save([test_mask, test_img_tensor], './colab_demo/test_image.pth')

xinzi2018 commented

How can you create CelebA-HQ mask?

inbarhub commented

Hi, Can you supply the images from celebA you edit in the paper?

hxy-123-coder commented

@xinzi2018 Hello,do you solve this problem?