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

Switched output saving to numpy format

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-03-21 17:38:22 +01:00
parent 69cc07ed12
commit ae109fd131
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -154,12 +154,13 @@ def predict(dataset: tf.data.Dataset,
outputs = _predict_one_epoch(dataset, use_dropout, forward_passes_per_image, **checkpointables) outputs = _predict_one_epoch(dataset, use_dropout, forward_passes_per_image, **checkpointables)
# save predictions # save predictions
filename = 'ssd_predictions.bin' filename = 'ssd_predictions.npy'
if use_dropout: if use_dropout:
filename = 'dropout-' + filename filename = 'dropout-' + filename
output_file = os.path.join(output_path, filename)
with open(output_path + filename, 'wb') as file: with open(output_file, 'wb') as file:
pickle.dump(outputs, file) np.save(file, outputs['decoded_predictions'], allow_pickle=False, fix_imports=False)
if verbose: if verbose:
print(( print((