From: will <greenwoodw50@gmail.com>
Date: Tue, 10 Sep 2024 16:27:32 +0000 (+0100)
Subject: Changed loop.py to chache spectrograms
X-Git-Url: https://git.ozva.co.uk/?a=commitdiff_plain;h=ab663e2e4a692487dfc24eb6e845f070f3aed4e8;p=audio-over-stft

Changed loop.py to chache spectrograms
---

diff --git a/loop.py b/loop.py
index e0c534a..34110ed 100755
--- a/loop.py
+++ b/loop.py
@@ -90,6 +90,8 @@ def process_loop(
 	print("caching data...")
 	caching = True
 
+	all_spectrums = np.zeros((segment_count, window_height, window_size, 3), dtype=np.uint8)
+
 	for segment_index in tqdm(range(segment_count)):
 		# get the specturm of the current sample
 		segment_start = segment_index * segment_samples
@@ -108,7 +110,8 @@ def process_loop(
 		if segment_index == 10:
 			cv.imwrite("spectrum_sample.jpg", spectrum)
 
-		np.save(f"cache/frame{segment_index}.npy", spectrum)
+		all_spectrums[segment_index] = spectrum
+		#np.save(f"cache/frame{segment_index}.npy", spectrum)
 
 	segment_index = 0
 	print("cached!")
@@ -116,7 +119,7 @@ def process_loop(
 
 	while segment_index < segment_count:
 
-		spectrum = np.load(f"cache/frame{segment_index}.npy")
+		spectrum = all_spectrums[segment_index]#np.load(f"cache/frame{segment_index}.npy")
 
 		# display and capture
 		camera.display(spectrum)
@@ -171,9 +174,12 @@ notes:
 
 # define parameters
 sample_rate = 22_050
-window_size = 150
+# 64 81 100 121 144 169 192 225
+window_size = 144
 window_height = 80
 
+hop_size = window_size // 2
+
 caching = False
 
 if __name__ == "__main__":