admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed bug with hw accel ffmpeg encoders by testing them beforehand #12. Added command arg —colab_performance to maximize performance in Google Colab. Colab notebook updated.

Browse Source
This commit is contained in:
Xavi Vinaixa
2023-06-07 17:22:27 +02:00
parent ef9b64ad92
commit 5bc01cf583
4 changed files with 74 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app.py
View File

@@ -6,9 +6,10 @@ parser = argparse.ArgumentParser(description='Refacer')
parser.add_argument("--max_num_faces", help="Max number of faces on UI", default=5)
parser.add_argument("--force_cpu", help="Force CPU mode", default=False,action="store_true")
parser.add_argument("--share_gradio", help="Share Gradio", default=False,action="store_true")
parser.add_argument("--colab_performance", help="Use in colab for better performance", default=False,action="store_true")
args = parser.parse_args()
refacer = Refacer(force_cpu=args.force_cpu)
refacer = Refacer(force_cpu=args.force_cpu,colab_performance=args.colab_performance)
num_faces=args.max_num_faces