Adding changes required for furthur editing - 2
This commit is contained in:
parent
c7fddcaae4
commit
c339f4dd6e
5
Utils.py
5
Utils.py
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user