search

genforce / mganprior

[CVPR 2020] Image Processing Using Multi-Code GAN Prior
https://genforce.github.io/mganprior/
289 stars 44 forks source link

pretrain weights #14

Closed lkostakis closed 3 years ago

lkostakis commented 3 years ago

I have downloaded the official weights for PGGAN and saved the file in /mganprior/pretrain/tensorflow/karras2018iclr-lsun-bedroom-256x256.pkl. When I execute: 

python inpainting.py
       --gan_model pggan_bedroom
       --target_images ./examples/inpainting/bedroom
       --outputs ./inpainting
       --mask ./examples/masks/mask-1.png
       --composing_layer 4
       --z_number 30

I get "[2021-03-08 23:47:56,814][WARNING] No pre-trained weights will be loaded!". Anyone had same problem ?

lkostakis commented 3 years ago

I figured it, was wrong path, I should save in /models/pretrain/tensorflow/...