From 237e0c13e77f333c075f6c744d1fbc6dc8a3316c Mon Sep 17 00:00:00 2001 From: Xintao Date: Thu, 20 May 2021 10:23:42 +0800 Subject: [PATCH] update gfpgan model --- models/gfpgan_model.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/models/gfpgan_model.py b/models/gfpgan_model.py index 3c92d2e..4716be4 100644 --- a/models/gfpgan_model.py +++ b/models/gfpgan_model.py @@ -202,25 +202,25 @@ class GFPGANModel(BaseModel): if 'gt' in data: self.gt = data['gt'].to(self.device) - # uncomment to check data - # import torchvision - # if self.opt['rank'] == 0: - # import os - # os.makedirs('tmp/gt', exist_ok=True) - # os.makedirs('tmp/lq', exist_ok=True) - # print(self.idx) - # torchvision.utils.save_image( - # self.gt, f'tmp/gt/gt_{self.idx}.png', nrow=4, padding=2, normalize=True, range=(-1, 1)) - # torchvision.utils.save_image( - # self.lq, f'tmp/lq/lq{self.idx}.png', nrow=4, padding=2, normalize=True, range=(-1, 1)) - # self.idx = self.idx + 1 - if 'loc_left_eye' in data: # get facial component locations, shape (batch, 4) self.loc_left_eyes = data['loc_left_eye'] self.loc_right_eyes = data['loc_right_eye'] self.loc_mouths = data['loc_mouth'] + # uncomment to check data + # import torchvision + # if self.opt['rank'] == 0: + # import os + # os.makedirs('tmp/gt', exist_ok=True) + # os.makedirs('tmp/lq', exist_ok=True) + # print(self.idx) + # torchvision.utils.save_image( + # self.gt, f'tmp/gt/gt_{self.idx}.png', nrow=4, padding=2, normalize=True, range=(-1, 1)) + # torchvision.utils.save_image( + # self.lq, f'tmp/lq/lq{self.idx}.png', nrow=4, padding=2, normalize=True, range=(-1, 1)) + # self.idx = self.idx + 1 + def construct_img_pyramid(self): pyramid_gt = [self.gt] down_img = self.gt