1 Star 0 Fork 1

RudonOpenSource/faceswap

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
faceswap.py 1.86 KB
一键复制 编辑 原始数据 按行查看 历史
torzdf 提交于 2023-06-27 11:27 . Rebase code (#1326)
#!/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()
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/RudonOpenSource/faceswap.git
git@gitee.com:RudonOpenSource/faceswap.git
RudonOpenSource
faceswap
faceswap
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385