researchmm/AOT-GAN-for-Inpainting

Loading pre-trained model throws error

Closed this issue · 3 comments

karynaur commented

After adding the downloaded pretrained model into experiments/, running demo.py throws

IsADirectoryError: [Errno 21] Is a directory: 'experiments/'

image

zengyh1900 commented

Hi @karynaur,

Thanks for your interest!
For the issue, you need to download the pre-trained model and put it under the folder experiments.
Then specify the path to the model by --pre_train experiments/xxx/xxx.pt.

Thanks.

karynaur commented

Thank you for the reply @1900zyh !!
I tried the same and it seems to run but without any output

!python demo.py --dir_image ../1.jpg --pre_train experiments/G0000000.pt --painter bbox --outputs /content/out.jpg

Like in here the cell runs successfully but i dont seem to find any image out.jpg in /content/

karynaur commented

Fixed it! Thanks!
This solved it:

>>>python demo.py --dir_image ./images/ --pre_train ./experiments/places/G0000000.pt
  • 👍1