diff --git a/background_subtraction_heat.py b/background_subtraction_heat.py index a512211..d712c07 100644 --- a/background_subtraction_heat.py +++ b/background_subtraction_heat.py @@ -82,6 +82,8 @@ if __name__ == '__main__': grouper = RecordingGrouper(data) groups = grouper.getRecordingGroups() + out = cv2.VideoWriter('outpy.avi',cv2.VideoWriter_fourcc('M','J','P','G'), 30, (512, 424), isColor=False) + for group in groups: heatmaps = [] for recording in group[1]: @@ -95,6 +97,7 @@ if __name__ == '__main__': converted = to_8_bit_image(frame, display_min=100, display_max=8000) heatmap.update(converted) cv2.imshow("IR", converted) + # out.write(converted) cv2.imshow("Heatmap", heatmap.heatmap) cv2.moveWindow("Heatmap", 600, 100) if cv2.waitKey(1) & 0xFF == ord('q'): @@ -110,5 +113,7 @@ if __name__ == '__main__': cv2.waitKey() cv2.destroyAllWindows() + # out.release() + # cv2.destroyAllWindows()