v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-18 08:10:16 -07:00
Code Issues Projects Releases Wiki Activity

update gfpgan model

Browse Source
This commit is contained in:
Xintao 2021-05-20 10:23:42 +08:00
parent f565932fb0
commit 237e0c13e7
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
models/gfpgan_model.py
View File

@ -202,25 +202,25 @@ class GFPGANModel(BaseModel):
if 'gt' in data: if 'gt' in data:
self.gt = data['gt'].to(self.device) 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: if 'loc_left_eye' in data:
# get facial component locations, shape (batch, 4) # get facial component locations, shape (batch, 4)
self.loc_left_eyes = data['loc_left_eye'] self.loc_left_eyes = data['loc_left_eye']
self.loc_right_eyes = data['loc_right_eye'] self.loc_right_eyes = data['loc_right_eye']
self.loc_mouths = data['loc_mouth'] 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): def construct_img_pyramid(self):
pyramid_gt = [self.gt] pyramid_gt = [self.gt]
down_img = self.gt down_img = self.gt