Could you apply this to other datasets like VGGface2?

[Jaep0805]

Could you please provide code to apply this to datasets other than CelebAHQ.
For example, I would like to apply this to VGGface2 to make a natural occluded VGGFace2 dataset

Thank you!