thermal data visualization

Author: smerdov

Analysis/ThermalDataVisualization.py

@@ -0,0 +1,28 @@
+import matplotlib.pyplot as plt
+import pandas as pd
+import numpy as np
+import os
+from config import *
+
+# plt.interactive(False)
+
+thermal_images_path = '../AndroidAppData/'
+filenames = os.listdir(thermal_images_path)
+
+for filenum, filename in enumerate(filenames):
+    try:
+        thermal_data = pd.read_csv(thermal_images_path + filename, header=None, sep=';').values
+        # thermal_data.astype(float)
+        thermal_data = thermal_data.T
+
+        # plt.close()
+        plt.close()
+        fig, ax = plt.subplots()
+        ax.imshow(thermal_data, vmax=40, vmin=30, cmap='Blues')
+        fig.show()
+        fig.savefig(f'{pic_folder}thermal_image_{filenum}.png')
+    except:
+        print(f'Can\'t read {filename}')
+
+
+