From 17e9208c218c7ab5010823186589ba4bca401375 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Wed, 22 May 2019 14:02:59 +0200
Subject: [PATCH] Added removal of detections after no longer being used

Signed-off-by: Jim Martens <github@2martens.de>
---
 src/twomartens/masterthesis/ssd.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py
index de7813a..1e0bda9 100644
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@@ -201,6 +201,8 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
                     detections[i].extend(batch_item)
             
             observations = np.asarray(_get_observations(detections))
+            del detections
+            
             observations = ssd_output_decoder.decode_detections_fast(observations)
             result_transformed = []
             for i in range(batch_size):