yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #251 from valrus/allow-use-without-multiprocessing

Browse Source 
Add an option allowing use without a multiprocessing pool...
This commit is contained in:
Félix Voituret
2020-01-30 12:01:16 +01:00
committed by GitHub
parent dde9bc610e aff7f83a13
commit d5d372dd0f
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
spleeter/separator.py
View File

@@ -35,7 +35,7 @@ __license__ = 'MIT License'
class Separator(object): class Separator(object):
""" A wrapper class for performing separation. """ """ A wrapper class for performing separation. """
def __init__(self, params_descriptor, MWF=False): def __init__(self, params_descriptor, MWF=False, multiprocess=True):
""" Default constructor. """ Default constructor.
:param params_descriptor: Descriptor for TF params to be used. :param params_descriptor: Descriptor for TF params to be used.
@@ -45,7 +45,7 @@ class Separator(object):
self._sample_rate = self._params['sample_rate'] self._sample_rate = self._params['sample_rate']
self._MWF = MWF self._MWF = MWF
self._predictor = None self._predictor = None
self._pool = Pool() self._pool = Pool() if multiprocess else None
self._tasks = [] self._tasks = []
def _get_predictor(self): def _get_predictor(self):
@@ -136,12 +136,15 @@ class Separator(object):
f'Separated source path conflict : {path},' f'Separated source path conflict : {path},'
'please check your filename format')) 'please check your filename format'))
generated.append(path) generated.append(path)
task = self._pool.apply_async(audio_adapter.save, ( if self._pool:
path, task = self._pool.apply_async(audio_adapter.save, (
data, path,
self._sample_rate, data,
codec, self._sample_rate,
bitrate)) codec,
self._tasks.append(task) bitrate))
if synchronous: self._tasks.append(task)
else:
audio_adapter.save(path, data, self._sample_rate, codec, bitrate)
if synchronous and self._pool:
self.join() self.join()