def entrypoint(arguments, params):
""" Command entrypoint.
:param arguments: Command line parsed argument as argparse.Namespace.
:param params: Deserialized JSON configuration file provided in CLI args.
"""
# TODO: check with output naming.
print('entrypoint')
print('inputs args:')
print(arguments.inputs)
print('audio adapter arguments:')
print(arguments.audio_adapter)
print('arguments config:')
print(arguments.configuration)
print('MWF arguments')
print(arguments.MWF)
print('stft backend arguments')
print(arguments.stft_backend)
audio_adapter = get_audio_adapter(arguments.audio_adapter)
print(audio_adapter)
print(audio_adapter.__dict__)
separator = Separator(arguments.configuration,
MWF=arguments.MWF,
stft_backend=arguments.stft_backend)
print(separator.__dict__)
for filename in arguments.inputs:
print('Todos los argumentos de separate_to_file:')
print(filename)
print(arguments.output_path)
print(audio_adapter)
print(arguments.offset)
print(arguments.duration)
print(arguments.codec)
print(arguments.bitrate)
print(arguments.filename_format)
separator.separate_to_file(filename,
arguments.output_path,
audio_adapter=audio_adapter,
offset=arguments.offset,
duration=arguments.duration,
codec=arguments.codec,
bitrate=arguments.bitrate,
filename_format=arguments.filename_format,
synchronous=False)
separator.join()
if to_int:
wav = np.array(wav, dtype='int16')
return wav
p = pyaudio.PyAudio()
CHUNK = 2048
stream = p.open(format=pyaudio.paInt16,
channels=1,
rate=8000,
frames_per_buffer=CHUNK,
input=True,
output=True)
infer = Separator()
while stream.is_active():
output = []
wav = np.fromstring(stream.read(CHUNK, exception_on_overflow=False),
np.int16)
print(wav.shape)
spec = stft(wav, False).T
spec_pred = stft(wav, True).T
for t in range(spec.shape[0]):
pred = infer.predict(spec_pred[t])
for i in range(129):
if pred[i] > 0.75:
spec[t][i] *= pred[i]
elif pred[i] > 0.5:
spec[t][i] *= 0.1
else:
def add_separator_vertical(self, position):
separator = Separator()
separator.position = position
separator.direction = VERTICAL
self.separators.append(separator)
self.handler.control.append(separator.control)
def add_separator_horizontal(self, position):
separator = Separator()
separator.position = position
separator.direction = HORIZONTAL
self.separators.append(separator)
self.handler.control.append(separator.control)
# -*- coding: utf-8 -*-
from separator import Separator
if __name__ == '__main__':
Separator(True)
def setUp(self):
self.x = Separator()
