Browse Source

load GFPGAN once and keep it in memory for #53

AUTOMATIC 3 years ago
parent
commit
d7b67d9b40
1 changed files with 9 additions and 1 deletions
  1. 9 1
      webui.py

+ 9 - 1
webui.py

@@ -179,8 +179,16 @@ def gfpgan_model_path():
     return found[0]
     return found[0]
 
 
 
 
+loaded_gfpgan_model = None
+
+
 def gfpgan():
 def gfpgan():
-    return GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+    global loaded_gfpgan_model
+
+    if loaded_gfpgan_model is None:
+        loaded_gfpgan_model = GFPGANer(model_path=gfpgan_model_path(), upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+
+    return loaded_gfpgan_model
 
 
 def gfpgan_fix_faces(gfpgan_model, np_image):
 def gfpgan_fix_faces(gfpgan_model, np_image):
     np_image_bgr = np_image[:, :, ::-1]
     np_image_bgr = np_image[:, :, ::-1]