mirror of
https://github.com/zebrajr/faceswap.git
synced 2025-12-06 12:20:27 +01:00
34b558426e
- preview tool - Prevent tkinter variables from exiting main thread
refactor:
- preview tool - Split module to smaller sub-modules + docs, locales
Typing:
- tools.preview.cli
Unit test:
- tools.preview.viewer
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
""" Command Line Arguments for tools """
|
|
import gettext
|
|
from typing import Any, List, Dict
|
|
|
|
from lib.cli.args import FaceSwapArgs
|
|
from lib.cli.actions import DirOrFileFullPaths, DirFullPaths, FileFullPaths
|
|
|
|
# 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, 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(dict(
|
|
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(dict(
|
|
opts=("-al", "--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(dict(
|
|
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(dict(
|
|
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
|