Can I train RestoreFormer on 1024x1024?

wzhouxiff / RestoreFormer

[CVPR 2022] RestoreFormer: High-Quality Blind Face Restoration from Undegraded Key-Value Pairs

Apache License 2.0

332 stars 35 forks source link

Can I train RestoreFormer on 1024x1024? #28

Open luser350 opened 4 months ago

luser350 commented 4 months ago

Hi, I want to train a model on 1k resolution. Since it is mentioned in the readme that FFHQ images were resized to 512x512. Does your model support 1K resolution training and generating 1K resolution images?

luser350 commented 4 months ago

@wzhouxiff can you please answer?

wzhouxiff commented 3 months ago

Yes, you can finetune the model of our RestoreFormer/ RestoreFormer++ on 1k resolution. However, we suggest adapting the HQ_Dictionary to 1k resolution first before restoration.