Customize masking

Thank you for your valuable work.

In your code, it is designed to generate random mask based on embedded input image.

But, I want to make code to customize mask region in image-level, not embedding-level.

How can I edit your code?