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

Where is the mask? #1

Closed aleksandereiken closed 2 years ago

aleksandereiken commented 2 years ago

Hi, The script is failing in line 31 of the main.py Traceback (most recent call last): File "main.py", line 31, in assert image.shape == mask.shape AttributeError: 'NoneType' object has no attribute 'shape'

Looks like there is no folder file "mask.png" in the github repo?

zuruoke commented 2 years ago

fixed!! let me know when it's working for you

P.S: preferable use istock pics of 683 x 1024 or aspect ratio 1.4 to get better results

Prem95 commented 2 years ago

Just change the path to an absolute path directing to your mask.png

Prem95 commented 2 years ago

It’s under utils/istock/landscape. Please check through the repo throughly before creating an issue

utils/istock/landscape/mask.png