How to mask the face during training?

[YoucanBaby]

Thx for your project!~

I find this sentence in readme: 'Implement better dreambooth training (like masking the face during training)'.

But how to mask the face during training? Eg: random mask 50%?

[KohakuBlueleaf]

Thx for your project!~

I find this sentence in readme: 'Implement better dreambooth training (like masking the face during training)'.

But how to mask the face during training? Eg: random mask 50%?

I don't know how to do it either I just know there is something like that I have no time to read others' source code(

That's why it is in TODO but not Features

[levi]

Check out the face mask preprocessing of this PTI repo: https://github.com/replicate/cog-sdxl/blob/main/preprocess.py

[KohakuBlueleaf]

@levi Thx for informations now I know what it does.

[YoucanBaby]

Check out the face mask preprocessing of this PTI repo: https://github.com/replicate/cog-sdxl/blob/main/preprocess.py

Thx for your code~!