def test_set_output_dir(self):
path = "./output"
environment.set_output_dir(path)
abs_path = os.path.abspath(path)
assert environment.OUTPUT_DIR == abs_path
assert os.environ["OUTPUT_DIR"] == abs_path
assert environment._EXPERIMENT_DIR == os.path.join(
abs_path,
"{experiment_id}",
)
assert environment._TENSORBOARD_DIR == os.path.join(
abs_path,
"{experiment_id}",
"tensorboard",
)
assert environment._CHECKPOINTS_DIR == os.path.join(
abs_path,
"{experiment_id}",
"checkpoints",
)
assert environment.EXPERIMENT_DIR is None
assert environment.TENSORBOARD_DIR is None
assert environment.CHECKPOINTS_DIR is None
def train(config, experiment_id, recreate, profile_step, data_dir, output_dir):
environment.set_data_dir(data_dir)
environment.set_output_dir(output_dir)
experiment_id, checkpoint_name = run_train(config, experiment_id, recreate,
profile_step)
click.echo('Next step: blueoil convert -e {} -p {}'.format(
experiment_id, checkpoint_name))
def convert(experiment_id, checkpoint, template, image_size, project_name, output_dir, test_image):
environment.set_output_dir(output_dir)
export_output_root_dir = run_convert(
experiment_id, checkpoint, template, image_size, project_name, test_image,
)
click.echo('Output files are generated in {}'.format(export_output_root_dir))
click.echo('Please see {}/README.md to run prediction'.format(export_output_root_dir))
def test_set_output_dir_after_init(self):
experiment_id = "experiment001"
environment.init(experiment_id)
path = "./output"
environment.set_output_dir(path)
abs_path = os.path.abspath(path)
assert environment.EXPERIMENT_DIR == os.path.join(
abs_path,
experiment_id,
)
assert environment.TENSORBOARD_DIR == os.path.join(
abs_path,
experiment_id,
"tensorboard",
)
assert environment.CHECKPOINTS_DIR == os.path.join(
abs_path,
experiment_id,
"checkpoints",
)
def predict(input, output, experiment_id, config, checkpoint, save_images, output_dir):
environment.set_output_dir(output_dir)
run_predict(input, output, experiment_id, config, checkpoint, save_images)
click.echo('Result files are created: {}'.format(output))
def test_set_output_dir_with_gcs_path(self):
path = "gs://output"
environment.set_output_dir(path)
assert environment.OUTPUT_DIR == path
assert os.environ["OUTPUT_DIR"] == path