how to make my own dataset

[yangyuke001]

Hello, thank you for your excellent work！ May I ask if I want to train a model specifically for removing watermarks? The background images are all product images from e-commerce platforms. My current idea is as follows: I plan to collect a certain amount of e-commerce image dataset, each image without watermarks, randomly generate masks using the author's methods, and further create a training set. Please help me check if my idea is correct?

[LikeGiver]

I'm also interested in this idea, my current model is based on this project: https://github.com/lcy0604/EraseNet , but the effect is not satisfactory. I think it is due to the small number of model parameters.