def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): trainer_params, scenario = Trainer.parse_trainer_params(values) parser.add_root_argument("trainer", trainer_params.__class__, default=trainer_params) setattr(namespace, "scenario_cls", scenario)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): scenario, scenario_params = ScenarioBase.from_path(values) evaluator_params = scenario_params.evaluator parser = TFAIPArgumentParser() parser.add_root_argument("evaluator", evaluator_params.__class__, evaluator_params) setattr(namespace, self.dest, scenario) setattr(namespace, self.dest + "_params", scenario_params)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): from tfaip.scenario.scenariobase import import_scenario scenario = import_scenario(values) # Now pass the real args of the scenario default_trainer_params = scenario.default_trainer_params() parser.add_root_argument("trainer", default_trainer_params.__class__, default=default_trainer_params) setattr(namespace, self.dest, scenario)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): from tfaip.imports import ScenarioBase export_dir = values scenario, scenario_params = ScenarioBase.from_path(export_dir) parser.add_root_argument("scenario", scenario_params.__class__, default=scenario_params) setattr(namespace, self.dest, values) setattr(namespace, "scenario_cls", scenario)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): from tfaip.imports import Trainer output_dir = values trainer_params, scenario = Trainer.parse_trainer_params(output_dir) # parse additional args parser.add_root_argument("trainer", trainer_params.__class__, default=trainer_params) setattr(namespace, self.dest, values) setattr(namespace, "scenario_cls", scenario)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): scenario, scenario_params = ScenarioBase.from_path(values[0]) predict_params = scenario.predictor_cls().params_cls()() parser.add_root_argument("data", scenario.predict_generator_params_cls()) parser.add_root_argument("predict", predict_params.__class__, default=predict_params) setattr(namespace, self.dest, values) setattr(namespace, "scenario", scenario) setattr(namespace, "scenario_params", scenario_params)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): from tfaip.imports import ScenarioBase export_dirs = values scenario, scenario_params = ScenarioBase.from_path( export_dirs[0]) # scenario based on first model lav_params = scenario.lav_cls().params_cls()() lav_params.model_path = export_dirs predictor_params = scenario.multi_predictor_cls().params_cls()() parser.add_root_argument("lav", scenario.lav_cls().params_cls(), default=lav_params) parser.add_root_argument("predictor", scenario.multi_predictor_cls().params_cls(), default=predictor_params, ignore=["pipeline"]), parser.add_root_argument("data", scenario.predict_generator_params_cls()) setattr(namespace, self.dest, values) setattr(namespace, "scenario", scenario) setattr(namespace, "scenario_params", scenario_params)
def __call__(self, parser: TFAIPArgumentParser, namespace, values, option_string=None): from tfaip.imports import ScenarioBase export_dir = values scenario, scenario_params = ScenarioBase.from_path(export_dir) default_gen_params = scenario.predict_generator_params_cls()() if os.path.exists(os.path.join(export_dir, "trainer_params.json")): # if trainer_params exist load val generator as default with open(os.path.join(export_dir, "trainer_params.json")) as f: p = scenario.trainer_cls().params_cls().from_json(f.read()) default_gen_params = p.gen.lav_gen()[0] lav_params = scenario.lav_cls().params_cls()() lav_params.model_path = export_dir parser.add_root_argument("data", DataGeneratorParams, default=default_gen_params) parser.add_root_argument("lav", scenario.lav_cls().params_cls(), default=lav_params) setattr(namespace, self.dest, values) setattr(namespace, "scenario", scenario) setattr(namespace, "scenario_params", scenario_params)