search

researchmm / AOT-GAN-for-Inpainting

[TVCG'2023] AOT-GAN for High-Resolution Image Inpainting (codebase for image inpainting)
https://arxiv.org/abs/2104.01431
Apache License 2.0
403 stars 64 forks source link

Loading pre-trained model throws error #1

Closed karynaur closed 3 years ago

karynaur commented 3 years ago

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

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

image

zengyh1900 commented 3 years ago

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 3 years ago

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 3 years ago

Fixed it! Thanks! This solved it:

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