search

zuruoke / watermark-removal

a machine learning image inpainting task that instinctively removes watermarks from image indistinguishable from the ground truth image
1.86k stars 291 forks source link

Are the pretrained models specific to the istock mask? #31

Open pixlmint opened 7 months ago

pixlmint commented 7 months ago

A lot of issues have been created asking for how to create a custom mask, and I've ran into the same problem.

I created a custom mask (png, black background, with just the watermark transparent) and put it into the correct directory. The script runs without issues, but the output image is exactly the same as the input image, just with some parameters changed (like it went from being 48-bit color to 24-bit color) so it's clear that something is being done. Just not the thing I'd like it to do.

So are the models provided through the google drive link just specific to the istock mask? And if so, how can we train them ourselves? Or should the models be universal?

cafuller commented 3 months ago

How did you create your custom mask image?