def __init__(self, arguments):
logger.debug("Initializing %s: (arguments: %s", self.__class__.__name__, arguments)
set_system_verbosity(arguments.loglevel)
self._update_type = arguments.processing
self._input_is_faces = arguments.input_type == "faces"
self._mask_type = arguments.masker
self._output_opts = dict(blur_kernel=arguments.blur_kernel, threshold=arguments.threshold)
self._output_type = arguments.output_type
self._output_full_frame = arguments.full_frame
self._output_suffix = self._get_output_suffix()
self._face_count = 0
self._skip_count = 0
self._update_count = 0
self._check_input(arguments.input)
self._saver = self._set_saver(arguments)
self._loader = ImagesLoader(arguments.input, load_with_hash=self._input_is_faces)
self._alignments = Alignments(os.path.dirname(arguments.alignments),
filename=os.path.basename(arguments.alignments))
self._extractor = self._get_extractor()
self._extractor_input_thread = self._feed_extractor()
logger.debug("Initialized %s", self.__class__.__name__)
def __init__(self, arguments):
logger.debug("Initializing %s: (arguments: '%s'",
self.__class__.__name__, arguments)
self.args = arguments
set_system_verbosity(self.args.loglevel)
self.alignments = self.load_alignments()
logger.debug("Initialized %s", self.__class__.__name__)
def __init__(self, arguments):
logger.debug("Initializing %s: (arguments: '%s'", self.__class__.__name__, arguments)
set_system_verbosity(arguments.loglevel)
self.config_tools = ConfigTools()
self.lock = Lock()
self.trigger_patch = Event()
self.root = tk.Tk()
self.scaling = self.get_scaling()
self.tk_vars = dict(refresh=tk.BooleanVar(), busy=tk.BooleanVar())
for val in self.tk_vars.values():
val.set(False)
self.display = FacesDisplay(256, 64, self.tk_vars)
self.samples = Samples(arguments, 5, self.display, self.lock, self.trigger_patch)
self.patch = Patch(arguments,
self.samples,
self.display,
self.lock,
self.trigger_patch,
self.config_tools,
self.tk_vars)
self.initialize_tkinter()
self.image_canvas = None
self.opts_book = None
self.cli_frame = None # cli frame holds cli options
logger.debug("Initialized %s", self.__class__.__name__)
def process(self):
""" Call the training process object """
logger.debug("Starting Training Process")
logger.info("Training data directory: %s", self.args.model_dir)
# TODO Move these args to config and remove these deprecation warnings
if hasattr(self.args,
"warp_to_landmarks") and self.args.warp_to_landmarks:
deprecation_warning(
"`-wl`, ``--warp-to-landmarks``",
additional_info="This option will be available within training "
"config settings (/config/train.ini).")
if hasattr(self.args, "no_augment_color") and self.args.no_flip:
deprecation_warning(
"`-nac`, ``--no-augment-color``",
additional_info="This option will be available within training "
"config settings (/config/train.ini).")
set_system_verbosity(self.args.loglevel)
thread = self.start_thread()
# queue_manager.debug_monitor(1)
err = self.monitor(thread)
self.end_thread(thread, err)
logger.debug("Completed Training Process")
def process(self):
""" Call the training process object """
print("Training data directory: {}".format(self.args.model_dir))
lvl = '0' if self.args.verbose else '2'
set_system_verbosity(lvl)
thread = self.start_thread()
if self.stop == True:
self.end_thread(thread)
def process(self):
""" Call the training process object """
logger.debug("Starting Training Process")
logger.info("Training data directory: %s", self.args.model_dir)
set_system_verbosity(self.args.loglevel)
thread = self.start_thread()
# queue_manager.debug_monitor(1)
err = self.monitor(thread)
self.end_thread(thread, err)
logger.debug("Completed Training Process")
def process(self):
""" Call the training process object """
logger.info("Training data directory: %s", self.args.model_dir)
set_system_verbosity(self.args.loglevel)
thread = self.start_thread()
if self.args.preview:
self.monitor_preview()
else:
self.monitor_console()
self.end_thread(thread)
def process(self):
""" Call the training process object """
logger.debug("Starting Training Process")
logger.info("Training data directory: %s", self.args.model_dir)
set_system_verbosity(self.args.loglevel)
thread = self.start_thread()
# queue_manager.debug_monitor(1)
err = self.monitor(thread)
self.end_thread(thread, err)
logger.debug("Completed Training Process")
def __init__(self, backend="GPU", loglevel="INFO"):
logger.debug("Initializing %s: (backend: %s, loglevel: %s)",
self.__class__.__name__, backend, loglevel)
set_system_verbosity(loglevel)
backend = backend.upper()
git_model_id = 10
model_filename = ["vggface2_resnet50_v2.h5"]
self.input_size = 224
# Average image provided in https://github.com/ox-vgg/vgg_face2
self.average_img = np.array([91.4953, 103.8827, 131.0912])
self.model = self.get_model(git_model_id, model_filename, backend)
logger.debug("Initialized %s", self.__class__.__name__)
def process(self):
""" Call the training process object """
print("Training data directory: {}".format(self.args.model_dir))
lvl = '0' if self.args.verbose else '2'
set_system_verbosity(lvl)
thread = self.start_thread()
if self.args.preview:
self.monitor_preview()
else:
self.monitor_console()
self.end_thread(thread)
def execute_script(self, arguments):
""" Performs final set up and launches the requested :attr:`_command` with the given
command line arguments.
Monitors for errors and attempts to shut down the process cleanly on exit.
Parameters
----------
arguments: :class:`argparse.Namespace`
The command line arguments to be passed to the executing script.
"""
set_system_verbosity(arguments.loglevel)
is_gui = hasattr(arguments, "redirect_gui") and arguments.redirect_gui
log_setup(arguments.loglevel, arguments.logfile, self._command, is_gui)
success = False
if self._command != "gui":
self._configure_backend(arguments)
try:
script = self._import_script()
process = script(arguments)
process.process()
success = True
except FaceswapError as err:
for line in str(err).splitlines():
logger.error(line)
except KeyboardInterrupt: # pylint: disable=try-except-raise
raise
except SystemExit:
pass
except Exception: # pylint: disable=broad-except
crash_file = crash_log()
logger.exception("Got Exception on main handler:")
logger.critical(
"An unexpected crash has occurred. Crash report written to '%s'. "
"You MUST provide this file if seeking assistance. Please verify you "
"are running the latest version of faceswap before reporting",
crash_file)
finally:
safe_shutdown(got_error=not success)
def __init__(self, arguments):
logger.debug("Initializing %s: (arguments: '%s'", self.__class__.__name__, arguments)
self.args = arguments
set_system_verbosity(self.args.loglevel)
self.alignments = self.load_alignments()
logger.debug("Initialized %s", self.__class__.__name__)
def set_verbosity(verbose):
""" Set the system output verbosity """
lvl = '0' if verbose else '2'
set_system_verbosity(lvl)
def set_verbosity(loglevel):
""" Set the system output verbosity """
set_system_verbosity(loglevel)
def set_verbosity(verbose):
""" Set the system output verbosity """
lvl = '0' if verbose else '2'
set_system_verbosity(lvl)
def set_verbosity():
""" Set the system output verbosity """
set_system_verbosity()
def set_verbosity(loglevel):
""" Set the system output verbosity """
set_system_verbosity(loglevel)
def __init__(self, arguments):
self.args = arguments
set_system_verbosity()
dest_format = self.get_dest_format()
self.alignments = AlignmentData(self.args.alignments_file, dest_format)
def __init__(self, arguments):
set_system_verbosity(arguments.loglevel)
self.root = FaceswapGui(arguments.debug)
