From ca25d90b1496a98d9eb2036e94d1da492f531bc2 Mon Sep 17 00:00:00 2001
From: Bart Moyaers <bart.moyaers@kuleuven.be>
Date: Fri, 6 Dec 2019 11:52:50 +0100
Subject: [PATCH] test_heatmap group_bgf

---
 heatmap_test.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/heatmap_test.py b/heatmap_test.py
index 2c33894..812a04e 100644
--- a/heatmap_test.py
+++ b/heatmap_test.py
@@ -27,17 +27,17 @@ if __name__ == '__main__':
     # out = cv2.VideoWriter('outpy.avi',cv2.VideoWriter_fourc\c('M','J','P','G'), 30, (512, 424), isColor=False)
     for group in groups:
         heatmaps = []
+        groupheatmap = BackgroundHeatmapFromGroup(group)
         for recording in group[1]:
             first = True
             frames = VideoLoader.extract_frames(recording)
-            bgh = BackgroundHeatmap(VideoLoader.extract_frames(recording))
             for frame in frames:
-                bgh.update(frame)
+                groupheatmap.update(frame)
                 cv2.imshow("IR", frame)
                 # out.write(converted)
                 # cv2.imshow("Heatmap", heatmap.heatmap)
-                cv2.imshow("Background filter", bgh.lastframe)
-                diff = bgh.bgf_diff
+                cv2.imshow("Background filter", groupheatmap.lastframe)
+                diff = groupheatmap.bgf_diff
 
                 # Erode dilate
                 kernel = np.ones((5,5),np.uint8)
@@ -51,7 +51,9 @@ if __name__ == '__main__':
                 cv2.moveWindow("BGF diff", 1200, 100)
                 if cv2.waitKey(1) & 0xFF == ord('q'):
                     break
-            heatmaps.append(bgh.heatmap)
+            heatmaps.append(groupheatmap.heatmap)
+            # Reset current heatmap and sum
+            groupheatmap.reset()
             cv2.destroyAllWindows()
 
         for i, bgh in enumerate(heatmaps):