Moved general purpose functions up

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-11 16:23:32 +02:00
parent d53a0cb6b4
commit 0d527cf26d
1 changed files with 8 additions and 8 deletions

View File

@ -366,6 +366,14 @@ def _pickle(filename: str, content: Any) -> None:
with open(filename, "wb") as file:
pickle.dump(content, file)
def _get_nr_batches(data_length: int, batch_size: int) -> int:
return int(math.floor(data_length / batch_size))
def _get_nr_digits(data_length: int, batch_size: int) -> int:
return math.ceil(math.log10(math.ceil(data_length / batch_size)))
def _ssd_evaluate_unbatch(glob_string: str) -> List[np.ndarray]:
@ -668,14 +676,6 @@ def _ssd_get_tensorboard_callback(args: argparse.Namespace, save_summaries_on_de
return tensorboard_callback
def _get_nr_batches(data_length: int, batch_size: int) -> int:
return int(math.floor(data_length / batch_size))
def _get_nr_digits(data_length: int, batch_size: int) -> int:
return math.ceil(math.log10(math.ceil(data_length / batch_size)))
def _ssd_train_call(args: argparse.Namespace, train_function: callable,
train_generator: Generator, nr_batches_train: int,
val_generator: Generator, nr_batches_val: int,