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

ValueError: Couldn't find 'checkpoint' file or checkpoints in given directory model/ #20

Open theroadfreak opened 1 year ago

theroadfreak commented 1 year ago

Hello, hope you are doing well.

Im getting this issue even when i have the "checkpoint" file in the model/ folder. No it does not have the .txt on it, and i tried moving the model/ folder around. Here is my folder structure:

watermar-removal/ │ .gitignore │ batch_test.py │ guided_batch_test.py │ image.png │ inpaint.yml │ inpaint_model.py │ inpaint_ops.py │ main.py │ preprocess_image.py │ README.md │ ├───model │ checkpoint │ txt.text │ ├───utils │ └───istock │ ├───landscape │ │ mask.png │ │ │ └───potrait │ mask.png │ └───pycache

I made the portrait version in photoshop, you can see it here if you want " https://we.tl/t-VyVXJeQFRf "

Using python 3.7.0 and pip version 23.1.2, in a virtual env i installed all the required packages. I had to change one of the lines for the code to work, its this line in preprocess_image.py " assert image.shape == preprocessed_mask_image.shape " Please write if you need more info.

Have a great rest of your day.