naver-ai / StyleMapGAN

Official pytorch implementation of StyleMapGAN (CVPR 2021)
https://www.youtube.com/watch?v=qCapNyRA_Ng
Other
458 stars 81 forks source link

How to edit my own images ? #17

Open hdxpmc opened 3 years ago

hdxpmc commented 3 years ago

Hello, We followed the guide to prepare_data for my own images, and generate lmdb file. When run testing: python generate.py --ckpt expr/checkpoints/celeba_hq_256_8x8.pt --mixing_type local_editing --test_lmdb data/celeba_hq/LMDB_test --local_editing_part nose It inform no image.

How do we test on my own image ?

blandocs commented 3 years ago

We already calculated celeba_hq_test_GT_sorted_pair.pkl for pairing two images that have similar poses. Then we use that file to run the local_editing in generate.py.

If you want to edit your own images, then please use the demo code. That is the easiest way. python demo.py --ckpt expr/checkpoints/celeba_hq_256_8x8.pt --dataset celeba_hq

wjlszhang commented 2 years ago

@blandocs I was trying demo, but I found issue when trying image from internet

图片

any hints to solve it?

Thanks