add GFPGAN model

2023-03-26 13:39:09 +08:00
parent d938f2da3c
commit e7c7896bfa
7 changed files with 171 additions and 1 deletions
--- a/lama_cleaner/plugins/realesrgan.py
+++ b/lama_cleaner/plugins/realesrgan.py
@@ -71,6 +71,7 @@ class RealESRGANUpscaler(BasePlugin):
        model_info = REAL_ESRGAN_MODELS[name]

        model_path = download_model(model_info["url"], model_info["model_md5"])
+        logger.info(f"RealESRGAN model path: {model_path}")

        self.model = RealESRGANer(
            scale=model_info["scale"],