how to make my own dataset

[杨宇克]

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?