Skip to content

Commit

Permalink
Merge branch 'staging' of https://github.com/deepfakes/faceswap into …
Browse files Browse the repository at this point in the history
…staging
  • Loading branch information
torzdf committed Sep 25, 2019
2 parents 48bd903 + e83819f commit 8ccb1f5
Show file tree
Hide file tree
Showing 3 changed files with 17 additions and 9 deletions.
7 changes: 3 additions & 4 deletions lib/sysinfo.py
Original file line number Diff line number Diff line change
Expand Up @@ -344,10 +344,9 @@ def full_info(self):
retval += ("{0: <20} {1}\n".format(key + ":", sys_info[key]))
retval += "\n=============== Pip Packages ===============\n"
retval += self.installed_pip
if not self.is_conda:
return retval
retval += "\n\n============== Conda Packages ==============\n"
retval += self.installed_conda
if self.is_conda:
retval += "\n\n============== Conda Packages ==============\n"
retval += self.installed_conda
retval += self.state_file
retval += "\n\n================= Configs =================="
retval += self.configs
Expand Down
7 changes: 3 additions & 4 deletions plugins/train/trainer/_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -333,12 +333,11 @@ def compile_sample(self, batch_size, samples=None, images=None):
num_images = min(batch_size, num_images) if batch_size is not None else num_images
logger.debug("Compiling samples: (side: '%s', samples: %s)", self.side, num_images)
images = images if images is not None else self.target
samples = [samples[0:num_images]] if samples is not None else [self.samples[0:num_images]]
retval = [samples[0:num_images]] if samples is not None else [self.samples[0:num_images]]
if self.use_mask:
retval = [tgt[0:num_images] for tgt in images]
retval.extend(tgt[0:num_images] for tgt in images)
else:
retval = [images[0:num_images]]
retval = samples + retval
retval.extend(images[0:num_images])
return retval

def compile_timelapse_sample(self):
Expand Down
12 changes: 11 additions & 1 deletion scripts/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from lib.keypress import KBHit
from lib.multithreading import MultiThread
from lib.queue_manager import queue_manager # noqa pylint:disable=unused-import
from lib.utils import get_folder, get_image_paths, set_system_verbosity
from lib.utils import get_folder, get_image_paths, set_system_verbosity, deprecation_warning
from plugins.plugin_loader import PluginLoader

logger = logging.getLogger(__name__) # pylint: disable=invalid-name
Expand Down Expand Up @@ -90,6 +90,16 @@ 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)
Expand Down

0 comments on commit 8ccb1f5

Please sign in to comment.