mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-17 05:21:10 +08:00
torzdf
837bc2d51d
* FaceSwap 3 (#1515) * Update extract pipeline * Update requirements + setup for nvidia * Remove allow-growth option * tf.keras to keras updates * lib.model.losses - Port + fix all loss functions for Keras3 * lib.model - port initializers, layers. normalization to Keras3 * lib.model.autoclip to Keras 3 * Update mixed precision layer storage * model file to .keras format * Restructure nn_blocks to initialize layers in __init__ * Tensorboard - Trainer: Add Torch compatible Tensorboard callbacks - GUI event reader remove TF dependency * Loss logging - Flush TB logs on save - Replace TB live iterator for GUI * Backup models on total loss drop rather than per side * Update all models to Keras3 Compat * Remove lib.model.session * Update clip ViT to Keras 3 * plugins.extract.mask.unet-dfl - Fix for Keras3/Torch backend * Port AdaBelief to Keras 3 * setup.py: - Add --dev flag for dev tool install * Fix Keras 3 syntax * Fix LR Finder for Keras 3 * Fix mixed precision switching for Keras 3 * Add more optimizers + open up config setting * train: Remove updating FS1 weights to FS2 models * Alignments: Remove support for legacy .json files * tools.model: - Remove TF Saved Format saving - Fix Backup/Restore + Nan-Scan * Fix inference model creation for Keras 3 * Preview tool: Fix for Keras3 * setup.py: Configure keras backend * train: Migration of FS2 models to FS3 * Training: Default coverage to 100% * Remove DirectML backend * Update setup for MacOS * GUI: Force line reading to UTF-8 * Remove redundant Tensorflow references * Remove redundant code * Legacy model loading: Fix TFLamdaOp scalar ops and DepthwiseConv2D * Add vertical offset option for training * Github actions: Add more python versions * Add python version to workflow names * Github workflow: Exclude Python 3.12 for macOS * Implement custom training loop * Fs3 - Add RTX5xxx and ROCm 6.1-6.4 support (#1511) * setup.py: Add Cuda/ROCm version select options * bump minimum python version to 3.11 * Switch from setup.cgf to pyproject.toml * Documentation: Update all docs to use automodapi * Allow sysinfo to run with missing packages + correctly install tk under Linux * Bugfix: dot naming convention in clip models * lib.config: Centralise globally rather than passing as object - Add torch DataParallel for multi-gpu training - GUI: Group switches together when generating cli args - CLI: Remove deprecated multi-character argparse args - Refactor: - Centralise tensorboard reading/writing + unit tests - Create trainer plugin interfaces + add original + distributed * Update installers
62 lines
2.1 KiB
Python
Executable File
62 lines
2.1 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"):
|
|
import ctypes
|
|
windll = ctypes.windll.kernel32
|
|
os.environ["LANG"] = locale.windows_locale[windll.GetUserDefaultUILanguage()]
|
|
|
|
from lib.cli import args as cli_args # pylint:disable=wrong-import-position
|
|
from lib.cli.args_train import TrainArgs # pylint:disable=wrong-import-position
|
|
from lib.cli.args_extract_convert import ConvertArgs, ExtractArgs # noqa:E501 pylint:disable=wrong-import-position
|
|
from lib.config import generate_configs # pylint:disable=wrong-import-position
|
|
from lib.system import System # pylint:disable=wrong-import-position
|
|
|
|
# LOCALES
|
|
_LANG = gettext.translation("faceswap", localedir="locales", fallback=True)
|
|
_ = _LANG.gettext
|
|
|
|
system = System()
|
|
system.validate_python()
|
|
|
|
_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()
|
|
ExtractArgs(subparser, "extract", _("Extract the faces from pictures or a video"))
|
|
TrainArgs(subparser, "train", _("Train a model for the two faces A and B"))
|
|
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()
|