Added removal of detections after no longer being used

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-22 14:02:59 +02:00
parent 11edf3772a
commit 17e9208c21
1 changed files with 2 additions and 0 deletions
--- 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):