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

Updated debug print

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 13:13:01 +02:00
parent 412f0ac453
commit c96783b384
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

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