test.py doesn't inpaint image

GouravWadhwa / Hypergraphs-Image-Inpainting

(WACV 2021) Hyperrealistic Image Inpainting with Hypergraphs

90 stars 15 forks source link

test.py doesn't inpaint image #9

Closed asdj32n12dk closed 3 years ago

asdj32n12dk commented 3 years ago

Hello, I just launch test.py and instead of inpainting - it just fill the mask with black. (It's happening because prediction_coarse magnitude it too low) Places365_val_00032545_00%-05%_02 27%_Places365_test_00176672

GouravWadhwa commented 3 years ago

Hi, Can you please explain what you are doing. Because I think you are not loading the pretrained model correctly. Please download the pretrained model and give the correct prefix while testing.

Thank you Gourav

asdj32n12dk commented 3 years ago

ohh, sorry, got it. Thanks!

I was using wrong path for checkpoint.restore() "pretrained_models/Random/places2_256x256_random_mask.index" instead of "pretrained_models/Random/places2_256x256_random_mask"