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

Added conversion to numpy array

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 13:14:24 +02:00
parent c96783b384
commit 3fd7ba7373
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -203,7 +203,7 @@ def predict_keras(generator: callable,
# do observation stuff
predictions = np.asarray(_get_observations(detections))
else:
predictions = ssd_model.predict_on_batch(x)
predictions = np.asarray(ssd_model.predict_on_batch(x))
print(predictions[:, :, [-2, -1]])
decoded_predictions_batch = ssd_output_decoder.decode_detections_fast(