search

OpenTalker / StyleHEAT

[ECCV 2022] StyleHEAT: A framework for high-resolution editable talking face generation
MIT License
627 stars 77 forks source link

Question about training the video calibrator. #24

Closed xiong-jie-y closed 1 year ago

xiong-jie-y commented 1 year ago

I'm trying to train video calibrator with my dataset.

I created the lmdb format dataset following your procedure and run this procedure.

bash bash/train_video_styleheat.sh

Then I got an error like this.

PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x7fc8f138de50>

It seems the lmdb dataset should contain align data from this code.

Could you help me with this align data? Maybe the script used to generate lmdb is different from the one you used?

FeiiYin commented 1 year ago

The code for aligning the video face can be seen here: https://github.com/FeiiYin/StyleHEAT/blob/main/utils/video_preprocess/align_face.py