zuruoke/watermark-removal

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

Opened this issue · 0 comments

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.