no .pickle file for /infer_images folder? about celebbasis HOT 3 CLOSED

1. The existing images under ./infer_images are not used for training our model. Our method doesn't need regularization images (although using regularization images like DreamBooth may generate better results?). Therefore, the 00_start_align.sh doesn't include any regularization images.
2. It seems that you are using the following training setting in ./config/stable-diffusion/aigc_id.yaml:

data:
  params:
    train:
      target: ldm.data.face_id.FaceIdDatasetStyleGAN3

rather than:

... ...
      target: ldm.data.face_id.FaceIdDatasetOneShot

The FaceIdDatasetStyleGAN3 requires using StyleGAN3 generated faces and a pikle file listing the absolute path of each image, which is not provided yet.
We'll make this setting work, release the complete files, and modify the code and documentation ASAP!

from celebbasis.

Nvm, I got it working yay!!! No need to worry about the .pickle file for /infer_images or stylegan3 now! ^_^

I don't know exactly which of these changes I made fixed the issue, and I'm too laxy to check, but I guess these show what I did to get it to work. I had the right target in the .yaml file actually, and since FaceIdDatasetOneShot inherits from FaceIdDatasetStyleGAN3

I still needed to adjust the latter child class.

I guess there is only one id because I'm just training myself?

But In the paper it's possible to train multiple people and have them appear separate/distinct in a generated photo (which most other AIGC methods can't do btw! so that's awesome)

Then, the other change I had to make was here I think? maybe not if the defaults are overwritten anyway, I'm not sure

I just added some print statements so I could better understand the logical flow of the _load_from_pickle() method

and here you can see the print statements in the output

from celebbasis.

1. We add an example using StyleGAN3 synthetic faces, 71da8d9.
2. If there is only one id, the specific ids should be [0] instead of [1]. In specific ids, 1 means the second ID, 12156a1.

from celebbasis.