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/parse_args.py
+++ b/lama_cleaner/parse_args.py
@@ -101,6 +101,14 @@ def parse_args():
        type=str,
        choices=RealESRGANModelNameList,
    )
+    parser.add_argument(
+        "--enable-gfpgan",
+        action="store_true",
+        help="Enable GFPGAN face restore",
+    )
+    parser.add_argument(
+        "--gfpgan-device", default="cpu", type=str, choices=["cpu", "cuda"]
+    )
    parser.add_argument(
        "--enable-gif",
        action="store_true",