2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Formatted saved ssd predictions to make sorting easier

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-29 11:23:22 +02:00
parent 198325883c
commit 3d56d69122
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/ssd.py
View File

@ -39,6 +39,7 @@ import time
from typing import Dict from typing import Dict
from typing import Optional from typing import Optional
import math
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from tensorflow.python.ops import summary_ops_v2 from tensorflow.python.ops import summary_ops_v2
@ -175,6 +176,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
# go through the data set # go through the data set
counter = 0 counter = 0
nr_digits = math.ceil(math.log10(len(dataset)))
for inputs in dataset: for inputs in dataset:
decoded_predictions_batch = [] decoded_predictions_batch = []
if use_dropout: if use_dropout:
@ -185,7 +187,7 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
decoded_predictions_batch.append(np.array(ssd(inputs))) decoded_predictions_batch.append(np.array(ssd(inputs)))
# save predictions batch-wise to prevent memory problems # save predictions batch-wise to prevent memory problems
with open(f"{output_file}-{counter:d}.npy", 'wb') as file: with open(f"{output_file}-{counter:{nr_digits}d}.npy", 'wb') as file:
np.save(file, decoded_predictions_batch, allow_pickle=False, fix_imports=False) np.save(file, decoded_predictions_batch, allow_pickle=False, fix_imports=False)
counter += 1 counter += 1