Spaces:
Build error
Build error
Commit
·
3d0b5bf
1
Parent(s):
4bc22ca
improved conversion
Browse files
utils.py
CHANGED
|
@@ -515,10 +515,13 @@ def segment_analysis(segment, sr, s1s2:list):
|
|
| 515 |
# RMS Energy
|
| 516 |
rms_energy = np.sqrt(np.mean(segment**2))
|
| 517 |
|
| 518 |
-
# Calculate
|
| 519 |
-
fft = np.fft.rfft(segment)
|
| 520 |
freqs = np.fft.rfftfreq(len(segment), d=1/sr)
|
| 521 |
-
|
|
|
|
|
|
|
|
|
|
| 522 |
|
| 523 |
s1_to_s2_duration = []
|
| 524 |
s2_to_s1_duration = []
|
|
|
|
| 515 |
# RMS Energy
|
| 516 |
rms_energy = np.sqrt(np.mean(segment**2))
|
| 517 |
|
| 518 |
+
# Calculate frequency spectrum and find dominant frequencies
|
| 519 |
+
fft = np.abs(np.fft.rfft(segment))
|
| 520 |
freqs = np.fft.rfftfreq(len(segment), d=1/sr)
|
| 521 |
+
# Focus on frequency range typical for heart sounds (20-200 Hz)
|
| 522 |
+
mask = (freqs >= 20) & (freqs <= 200)
|
| 523 |
+
dominant_freq_idx = np.argmax(fft[mask])
|
| 524 |
+
mean_frequency = freqs[mask][dominant_freq_idx]
|
| 525 |
|
| 526 |
s1_to_s2_duration = []
|
| 527 |
s2_to_s1_duration = []
|