Added debug print for predictions

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-09 13:00:23 +02:00
parent 38ce65d367
commit 412f0ac453
1 changed files with 1 additions and 0 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -204,6 +204,7 @@ def predict_keras(generator: callable,
            predictions = np.asarray(_get_observations(detections))
        else:
            predictions = ssd_model.predict_on_batch(x)
+            print(predictions)
        
        decoded_predictions_batch = ssd_output_decoder.decode_detections_fast(
            y_pred=predictions,