zuruoke/watermark-removal

Where is the mask?

aleksandereiken opened this issue · 3 comments

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?

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

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

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

utils/istock/landscape/mask.png