Commit d7b67d9b authored by AUTOMATIC's avatar AUTOMATIC

load GFPGAN once and keep it in memory for #53

parent e5fc7bb5
......@@ -179,8 +179,16 @@ def gfpgan_model_path():
return found[0]
loaded_gfpgan_model = None
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):
np_image_bgr = np_image[:, :, ::-1]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment