jono/selcal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed weirdness, was caused by bin selection

Browse Source
This commit is contained in:
Jono Targett 2024-05-27 00:24:16 +09:30
parent eb2926f525
commit 700f34e729
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/selcal-fft.py
View File

@ -72,16 +72,16 @@ for i in range(num_segments):
return np.array([centre_frequency - width, centre_frequency + width]) return np.array([centre_frequency - width, centre_frequency + width])
def magnitude_in_band(band): def magnitude_in_band(band):
return np.sum(normalized_magnitudes[int(band[0]):int(band[1])]) return np.sum(
normalized_magnitudes[int(band[0]/frequency_resolution):int(band[1]/frequency_resolution)]
)
scores = { scores = {
tone:decibels(magnitude_in_band(band(frequency))) tone:decibels(magnitude_in_band(band(frequency)))
for tone,frequency in TONES.items() for tone,frequency in TONES.items()
} }
print(scores) active_tones = find_top_two_keys(scores, 3.0)
active_tones = find_top_two_keys(scores)
if active_tones: if active_tones:
print(active_tones) print(active_tones)