mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-18 06:20:53 +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
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
""" Command Line Arguments for tools """
|
|
import gettext
|
|
import typing as T
|
|
|
|
from lib.cli.args import FaceSwapArgs
|
|
from lib.cli.actions import DirOrFileFullPaths, DirFullPaths, FileFullPaths
|
|
from lib.utils import get_module_objects
|
|
|
|
# pylint:disable=duplicate-code
|
|
# LOCALES
|
|
_LANG = gettext.translation("tools.preview", localedir="locales", fallback=True)
|
|
_ = _LANG.gettext
|
|
|
|
|
|
_HELPTEXT = _("This command allows you to preview swaps to tweak convert settings.")
|
|
|
|
|
|
class PreviewArgs(FaceSwapArgs):
|
|
""" Class to parse the command line arguments for Preview (Convert Settings) tool """
|
|
|
|
@staticmethod
|
|
def get_info() -> str:
|
|
""" Return command information
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
Top line information about the Preview tool
|
|
"""
|
|
return _("Preview tool\nAllows you to configure your convert settings with a live preview")
|
|
|
|
@staticmethod
|
|
def get_argument_list() -> list[dict[str, T.Any]]:
|
|
""" Put the arguments in a list so that they are accessible from both argparse and gui
|
|
|
|
Returns
|
|
-------
|
|
list[dict[str, Any]]
|
|
Top command line options for the preview tool
|
|
"""
|
|
argument_list = []
|
|
argument_list.append({
|
|
"opts": ("-i", "--input-dir"),
|
|
"action": DirOrFileFullPaths,
|
|
"filetypes": "video",
|
|
"dest": "input_dir",
|
|
"group": _("data"),
|
|
"required": True,
|
|
"help": _(
|
|
"Input directory or video. Either a directory containing the image files you wish "
|
|
"to process or path to a video file.")})
|
|
argument_list.append({
|
|
"opts": ("-a", "--alignments"),
|
|
"action": FileFullPaths,
|
|
"filetypes": "alignments",
|
|
"type": str,
|
|
"group": _("data"),
|
|
"dest": "alignments_path",
|
|
"help": _(
|
|
"Path to the alignments file for the input, if not at the default location")})
|
|
argument_list.append({
|
|
"opts": ("-m", "--model-dir"),
|
|
"action": DirFullPaths,
|
|
"dest": "model_dir",
|
|
"group": _("data"),
|
|
"required": True,
|
|
"help": _(
|
|
"Model directory. A directory containing the trained model you wish to process.")})
|
|
argument_list.append({
|
|
"opts": ("-s", "--swap-model"),
|
|
"action": "store_true",
|
|
"dest": "swap_model",
|
|
"default": False,
|
|
"help": _("Swap the model. Instead of A -> B, swap B -> A")})
|
|
return argument_list
|
|
|
|
|
|
__all__ = get_module_objects(__name__)
|