Skip to content

Commit

Permalink
no-flip & no-augment-color deprecation warnings
Browse files Browse the repository at this point in the history
  • Loading branch information
torzdf committed Sep 24, 2019
1 parent 54d5159 commit 5110315
Showing 1 changed file with 11 additions and 1 deletion.
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, "no_flip") and self.args.no_flip:
deprecation_warning("`-nf`, ``--no-flip``",
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 5110315

Please sign in to comment.