mirror of
https://github.com/zebrajr/faceswap.git
synced 2025-12-06 00:20:09 +01:00
1c081aea7d
* tools.alignments - add export job * plugins.extract: Update __repr__ for ExtractorBatch dataclass * plugins.extract: Initial implementation of external import plugins * plugins.extract: Disable lm masks on ROI alignment data import * lib.align: Add `landmark_type` property to AlignedFace and return dummy data for ROI Landmarks pose estimate * plugins.extract: Add centering config item for align import and fix filename mapping for images * plugins.extract: Log warning on downstream plugins on limited alignment data * tools: Fix plugins for 4 point ROI landmarks (alignments, sort, mask) * tools.manual: Fix for 2D-4 ROI landmarks * training: Fix for 4 point ROI landmarks * lib.convert: Average color plugin. Avoid divide by zero errors * extract - external: - Default detector to 'external' when importing alignments - Handle different frame origin co-ordinates * alignments: Store video extension in alignments file * plugins.extract.external: Handle video file keys * plugins.extract.external: Output warning if missing data * locales + docs * plugins.extract.align.external: Roll the corner points to top-left for different origins * Clean up * linting fix
80 lines
4.0 KiB
Python
80 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
The default options for the faceswap Import Alignments plugin.
|
|
|
|
Defaults files should be named <plugin_name>_defaults.py
|
|
Any items placed into this file will automatically get added to the relevant config .ini files
|
|
within the faceswap/config folder.
|
|
|
|
The following variables should be defined:
|
|
_HELPTEXT: A string describing what this plugin does
|
|
_DEFAULTS: A dictionary containing the options, defaults and meta information. The
|
|
dictionary should be defined as:
|
|
{<option_name>: {<metadata>}}
|
|
|
|
<option_name> should always be lower text.
|
|
<metadata> dictionary requirements are listed below.
|
|
|
|
The following keys are expected for the _DEFAULTS <metadata> dict:
|
|
datatype: [required] A python type class. This limits the type of data that can be
|
|
provided in the .ini file and ensures that the value is returned in the
|
|
correct type to faceswap. Valid data types are: <class 'int'>, <class 'float'>,
|
|
<class 'str'>, <class 'bool'>.
|
|
default: [required] The default value for this option.
|
|
info: [required] A string describing what this option does.
|
|
group: [optional]. A group for grouping options together in the GUI. If not
|
|
provided this will not group this option with any others.
|
|
choices: [optional] If this option's datatype is of <class 'str'> then valid
|
|
selections can be defined here. This validates the option and also enables
|
|
a combobox / radio option in the GUI.
|
|
gui_radio: [optional] If <choices> are defined, this indicates that the GUI should use
|
|
radio buttons rather than a combobox to display this option.
|
|
min_max: [partial] For <class 'int'> and <class 'float'> data types this is required
|
|
otherwise it is ignored. Should be a tuple of min and max accepted values.
|
|
This is used for controlling the GUI slider range. Values are not enforced.
|
|
rounding: [partial] For <class 'int'> and <class 'float'> data types this is
|
|
required otherwise it is ignored. Used for the GUI slider. For floats, this
|
|
is the number of decimal places to display. For ints this is the step size.
|
|
fixed: [optional] [train only]. Training configurations are fixed when the model is
|
|
created, and then reloaded from the state file. Marking an item as fixed=False
|
|
indicates that this value can be changed for existing models, and will override
|
|
the value saved in the state file with the updated value in config. If not
|
|
provided this will default to True.
|
|
"""
|
|
|
|
|
|
_HELPTEXT = (
|
|
"Import Detector options.\n"
|
|
"Imports a detected face bounding box from an external .json file.\n"
|
|
)
|
|
|
|
|
|
_DEFAULTS = {
|
|
"file_name": {
|
|
"default": "import.json",
|
|
"info": "The import file should be stored in the same folder as the video (if extracting "
|
|
"from a video file) or inside the folder of images (if importing from a folder of images)",
|
|
"datatype": str,
|
|
"choices": [],
|
|
"group": "settings",
|
|
"gui_radio": False,
|
|
"fixed": True,
|
|
},
|
|
"origin": {
|
|
"default": "top-left",
|
|
"info": "The origin (0, 0) location of the co-ordinates system used. "
|
|
"\n\t top-left: The origin (0, 0) of the canvas is at the top left "
|
|
"corner."
|
|
"\n\t bottom-left: The origin (0, 0) of the canvas is at the bottom "
|
|
"left corner."
|
|
"\n\t top-right: The origin (0, 0) of the canvas is at the top right "
|
|
"corner."
|
|
"\n\t bottom-right: The origin (0, 0) of the canvas is at the bottom "
|
|
"right corner.",
|
|
"datatype": str,
|
|
"choices": ["top-left", "bottom-left", "top-right", "bottom-right"],
|
|
"group": "output",
|
|
"gui_radio": True
|
|
}
|
|
}
|