Return onset times

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()
 

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])