From 1f1b9c62dec7cae001b79670139b3c2a84b73af4 Mon Sep 17 00:00:00 2001
From: Sting <loic.allegre@ensiie.fr>
Date: Wed, 21 Jun 2023 12:09:33 +0200
Subject: [PATCH] Return onset times
---
autokara.py | 2 +-
segment.py | 7 ++++++-
2 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/autokara.py b/autokara.py
index 894308d..850cd0e 100644
--- a/autokara.py
+++ b/autokara.py
@@ -29,5 +29,5 @@ vocals_file = "./media/vocals/htdemucs/%s/vocals.wav" % basename
seg = Segment(vocals_file)
-seg.onsets()
+onset_times = seg.onsets()
diff --git a/segment.py b/segment.py
index d5c2eda..91452de 100644
--- a/segment.py
+++ b/segment.py
@@ -27,7 +27,10 @@ class Segment:
rms = librosa.feature.rms(S=S)
onset_bt_rms = librosa.onset.onset_backtrack(onset_raw, rms[0])
- print(onset_bt_rms)
+ onset_bt_times = librosa.frames_to_time(onset_bt, sr=sr)
+ onset_bt_rms_times = librosa.frames_to_time(onset_bt_rms, sr=sr)
+
+ # print(onset_bt_rms_times)
'''
fig, ax = plt.subplots(nrows=3, sharex=True)
@@ -45,6 +48,8 @@ class Segment:
plt.show()
'''
+ return onset_bt_rms_times
+
if __name__ == "__main__":
seg = Segment(sys.argv[1])
--
GitLab