search

marcbelmont / cnn-watermark-removal

Fully convolutional deep neural network to remove transparent overlays from images
1.23k stars 227 forks source link

python watermarks.py --image assets/output.png --selection assets/cat-selection.png? #3

Closed FancyXun closed 6 years ago

FancyXun commented 6 years ago

what does the output.png look like when run python watermarks.py --image assets/output.png --selection assets/cat-selection.png? my output.png still has the watermark and it looks like a part of source image.

marcbelmont commented 6 years ago

The output should look like column 2 of https://github.com/marcbelmont/cnn-watermark-removal/blob/master/assets/example.png . For how many epochs did you train? Can you share your model weights?

FancyXun commented 6 years ago

No reminder to me for your answer... I can't find my model weights in /tmp. Maybe restarting computer made it deleted? my epochs is 1e5,the source is 1e6,but it is too long for me.Thanks for your reply @marcbelmont

marcbelmont commented 6 years ago

Yes, the content of /tmp is removed after a restart. You can use those weights: https://github.com/marcbelmont/cnn-watermark-removal/files/1594328/data.zip