CelebA-HQ pth missing

[ucalyptus2]

I could not find for CelebA-HQ in https://github.com/ermongroup/SDEdit/blob/main/functions/process_data.py
@junyanz

[aferriss]

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

[chenlin9]

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]

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

[sorayutmild]

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

[xinzi2018]

How can you create CelebA-HQ mask?

[inbarhub]

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

[hxy-123-coder]

@xinzi2018 Hello,do you solve this problem?