Adding changes required for furthur editing - 2

This commit is contained in:
parth aranke 2024-06-04 10:42:38 +05:30
parent c7fddcaae4
commit c339f4dd6e

View File

@ -84,9 +84,6 @@ class ProceesAudio():
print(len(y), len(x))
modified_segs = []
prev_gains = np.ones(10)
max_part_amp = 0
for segment, mod in zip(segments, y[0]):
_, audios = split_stereo(segment=segment, max_amp=max_amp, sr=sr, num_parts=10)
final_audios = []
@ -114,7 +111,7 @@ class ProceesAudio():
modified_segs = np.concatenate(modified_segs)
return modified_segs.astype(np.float32)
return modified_segs.astype(np.float32), y[0]
def get_training_data(self, file_path, data_dir):
for song in os.listdir(data_dir):