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

fix bug in inference: RealESRGAN model is None

Browse Source
This commit is contained in:
Xintao 2021-12-12 22:46:07 +08:00
parent 37237da798
commit ad1397180d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference_gfpgan.py
View File

@ -46,10 +46,13 @@ def main():
'If you really want to use it, please modify the corresponding codes.') 'If you really want to use it, please modify the corresponding codes.')
bg_upsampler = None bg_upsampler = None
else: else:
from basicsr.archs.rrdbnet_arch import RRDBNet
from realesrgan import RealESRGANer from realesrgan import RealESRGANer
model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=2)
bg_upsampler = RealESRGANer( bg_upsampler = RealESRGANer(
scale=2, scale=2,
model_path='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth', model_path='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth',
model=model,
tile=args.bg_tile, tile=args.bg_tile,
tile_pad=10, tile_pad=10,
pre_pad=0, pre_pad=0,