fix bug in inference: RealESRGAN model is None

xinntao · xinntao · commit ad1397180d23 · 2021-12-12T22:46:07.000+08:00
diff --git a/inference_gfpgan.py b/inference_gfpgan.py
@@ -46,10 +46,13 @@ def main():
                           'If you really want to use it, please modify the corresponding codes.')
             bg_upsampler = None
         else:
+            from basicsr.archs.rrdbnet_arch import RRDBNet
             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(
                 scale=2,
                 model_path='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth',
+                model=model,
                 tile=args.bg_tile,
                 tile_pad=10,
                 pre_pad=0,
