microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
15.11k stars 1.99k forks source link

Can't get global restoration to work. #185

Open acc3d opened 3 years ago

acc3d commented 3 years ago

I keep getting Mapping: You are using the mapping model without global restoration. despite having unzipped the checkpoints correctly. Any idea why?

zelenooki87 commented 3 years ago

Yes. This error is present also on Google Colab. /content/photo_restoration

Running Stage 1: Overall restoration

Mapping: You are using the mapping model without global restoration. ./checkpoints/restoration/VAE_A_quality/latest_net_G.pth not exists yet! ./checkpoints/restoration/VAE_B_quality/latest_net_G.pth not exists yet! ./checkpoints/restoration/mapping_quality/latest_net_mapping_net.pth not exists yet!

I hope, that devs will fix this issue.

semel1 commented 3 years ago

I have the same issue: "You are using the mapping model without global restoration."

tangger2000 commented 3 years ago

me,too

KeygenLLC commented 3 years ago

Same here, local install, get the same warning using the regular script or trying to run inside Global. Running CPU only so maybe that's an issue, but would like to know from devs if this is a known bug, lack of CUDA, or a setup problem.

ruanqizhen commented 2 years ago

got the same problem on linux

wilk3ns commented 2 years ago

Any updates yet? Running same issue on arch

Joeweav commented 2 years ago

I have the same issue here, running locally on Win 10 and CPU. I downloaded and installed all the files, and verified everything is in the proper place. One thing I did notice is one of the things they had me grab already existed but the version they said to get was newer so I got it. I may try pulling the repo and see if I can figure out what it was and put the original pieces back and see what it does

Here is a sample input and what the first stage does to it.. Any ideas?

input Phase_one_restore .

Baoanxia commented 2 years ago

me too.

shawtux commented 2 years ago

Anyone got this to work?

sirotaku commented 2 years ago

I have the same issue here. Running Stage 1: Overall restoration

Mapping: You are using the mapping model without global restoration. ./checkpoints/restoration/VAE_A_quality/latest_net_G.pth not exists yet! ./checkpoints/restoration/VAE_B_quality/latest_net_G.pth not exists yet! ./checkpoints/restoration/mapping_quality/latest_net_mapping_net.pth not exists yet!

jluyt123 commented 11 months ago

Was there any solution to this? I am getting the same things and some of my images are just blank, plus the results are sub-par.

alojal61 commented 10 months ago

I've had the same issue, I fix it copying the folder CHECKPOINT inside global_checkpoints/ to Global/

Tomtanium commented 8 months ago

Adding the '--checkpoint_name checkpoints' flag worked for me

muhammedmostafa commented 7 months ago

worked for me added ----checkpoints_dir checkpoints for --checkpoints_dir checkpoints , i see that every function has it's own argument name

Pete11 commented 2 months ago

Hi guys, I'm so frustrated with this. Can anyone tell as for a total noob what's the thing with this? I am so lost with all this. Where I should put all that files ? Thank you in advance!

Set device_ids as a comma separated list of device ids (i.e. 0 or 1,2). Use -1 for cpu.

vae_a Place in models/vae/.

restoration/VAE_A_quality/latest_net_G.pth vae_b Extract the following models and place them inside models/vae/.

restoration/VAE_B_quality/latest_net_G.pth restoration/VAE_B_scratch/latest_net_G.pth (scratch_detection) mapping_net Extract the following models and place them inside models/checkpoints/.

restoration/mapping_quality/latest_net_mapping_net.pth restoration/mapping_scratch/latest_net_mapping_net.pth (scratch_detection) restoration/mapping_Patch_Attention/latest_net_mapping_net.pth (mapping_patch_attention) Load Scratch Mask Model scratch_model Extract the following models and place them inside models/checkpoints/.

detection/FT_Epoch_latest.pt Face Detection Models (Stages 2-4) Load Face Detector Model (Dlib) Download - shape_predictor_68_face_landmarks.dat

Extract the following models and place them inside models/facedetection/ (custom directory).

shape_predictor_68_face_landmarks shape_predictor_68_face_landmarks.dat Load Face Enhancer Model Download - Face Enhancement Models

Extract the following models and place them inside models/checkpoints/.

Set device_ids as a comma separated list of device ids (i.e. 0 or 1,2). Use -1 for cpu.

face_enhance_model Setting_9_epoch_100/latest_net_G.pth (256x256) FaceSR_512/latest_net_G.pth (512x512)