mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 10:43:27 -04:00
6a3b674bef
* Remove tensorflow_probability requirement * setup.py - fix progress bars * requirements.txt: Remove pre python 3.9 packages * update apple requirements.txt * update INSTALL.md * Remove python<3.9 code * setup.py - fix Windows Installer * typing: python3.9 compliant * Update pytest and readthedocs python versions * typing fixes * Python Version updates - Reduce max version to 3.10 - Default to 3.10 in installers - Remove incompatible 3.11 tests * Update dependencies * Downgrade imageio dep for Windows * typing: merge optional unions and fixes * Updates - min python version 3.10 - typing to python 3.10 spec - remove pre-tf2.10 code - Add conda tests * train: re-enable optimizer saving * Update dockerfiles * Update setup.py - Apple Conda deps to setup.py - Better Cuda + dependency handling * bugfix: Patch logging to prevent Autograph errors * Update dockerfiles * Setup.py - Setup.py - stdout to utf-8 * Add more OSes to github Actions * suppress mac-os end to end test
56 lines
1.9 KiB
Python
Executable file
56 lines
1.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
""" The master faceswap.py script """
|
|
import gettext
|
|
import locale
|
|
import os
|
|
import sys
|
|
|
|
# Translations don't work by default in Windows, so hack in environment variable
|
|
if sys.platform.startswith("win"):
|
|
os.environ["LANG"], _ = locale.getdefaultlocale()
|
|
|
|
from lib.cli import args as cli_args # pylint:disable=wrong-import-position
|
|
from lib.config import generate_configs # pylint:disable=wrong-import-position
|
|
|
|
# LOCALES
|
|
_LANG = gettext.translation("faceswap", localedir="locales", fallback=True)
|
|
_ = _LANG.gettext
|
|
|
|
if sys.version_info < (3, 10):
|
|
raise ValueError("This program requires at least python 3.10")
|
|
|
|
_PARSER = cli_args.FullHelpArgumentParser()
|
|
|
|
|
|
def _bad_args(*args) -> None: # pylint:disable=unused-argument
|
|
""" Print help to console when bad arguments are provided. """
|
|
print(cli_args)
|
|
_PARSER.print_help()
|
|
sys.exit(0)
|
|
|
|
|
|
def _main() -> None:
|
|
""" The main entry point into Faceswap.
|
|
|
|
- Generates the config files, if they don't pre-exist.
|
|
- Compiles the :class:`~lib.cli.args.FullHelpArgumentParser` objects for each section of
|
|
Faceswap.
|
|
- Sets the default values and launches the relevant script.
|
|
- Outputs help if invalid parameters are provided.
|
|
"""
|
|
generate_configs()
|
|
|
|
subparser = _PARSER.add_subparsers()
|
|
cli_args.ExtractArgs(subparser, "extract", _("Extract the faces from pictures or a video"))
|
|
cli_args.TrainArgs(subparser, "train", _("Train a model for the two faces A and B"))
|
|
cli_args.ConvertArgs(subparser,
|
|
"convert",
|
|
_("Convert source pictures or video to a new one with the face swapped"))
|
|
cli_args.GuiArgs(subparser, "gui", _("Launch the Faceswap Graphical User Interface"))
|
|
_PARSER.set_defaults(func=_bad_args)
|
|
arguments = _PARSER.parse_args()
|
|
arguments.func(arguments)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
_main()
|