search

MrTornado24 / IDE-3D

[SIGGRAPH Asia 2022] IDE-3D: Interactive Disentangled Editing For High-Resolution 3D-aware Portrait Synthesis
https://mrtornado24.github.io/IDE-3D/
480 stars 44 forks source link

how to edit on my own images #18

Open flyingshan opened 1 year ago

flyingshan commented 1 year ago

I notice the advices from the section "Real portrait image editing", where first the following code should be executed:

python apps/infer_hybrid_encoder.py 
    --target_img /path/to/img_0.png
    --g_ckpt pretrained_models/ide3d-ffhq-64-512.pkl 
    --e_ckpt pretrained_models/encoder-base-hybrid.pkl
    --outdir out

But I found for my image, it does not have labels like those from dataset.json, so error occured:

File "/home/dianxin/hys/IDE-3D/apps/infer_hybrid_encoder.py", line 147, in c = [label_list[opts.target_img[-21:]]] KeyError: 'test.png'

How can I generate the labels for my own images? Thank you!

Bai426 commented 1 year ago

Yes, I found the same problem. The dataset.json is an empty file.