diff --git a/autokara.py b/autokara.py index 894308dfdeea9926de71caec176bd6c104a0a6c4..850cd0e3d3841a2a8c20c92563222863b74fd048 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 d5c2eda00a262980a44bf3611a44d6e6cd3a28d7..91452de420a89b528308d10f8ae351061d388b10 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])