yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix unknown encoder error

Browse Source
This commit is contained in:
Yusuke Goto
2020-01-01 19:31:13 +09:00
parent fc1e3d9a2f
commit 731102f24e
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
spleeter/audio/ffmpeg.py
View File

@@ -34,6 +34,15 @@ def _to_ffmpeg_time(n):
return '%d:%02d:%09.6f' % (h, m, s) return '%d:%02d:%09.6f' % (h, m, s)
def _to_ffmpeg_codec(codec):
ffmpeg_codecs = {
'm4a': 'aac',
'ogg': 'libvorbis',
'wma': 'wmav2',
}
return ffmpeg_codecs.get(codec) or codec
class FFMPEGProcessAudioAdapter(AudioAdapter): class FFMPEGProcessAudioAdapter(AudioAdapter):
""" An AudioAdapter implementation that use FFMPEG binary through """ An AudioAdapter implementation that use FFMPEG binary through
subprocess in order to perform I/O operation for audio processing. subprocess in order to perform I/O operation for audio processing.
@@ -112,7 +121,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
if bitrate: if bitrate:
output_kwargs['audio_bitrate'] = bitrate output_kwargs['audio_bitrate'] = bitrate
if codec is not None and codec != 'wav': if codec is not None and codec != 'wav':
output_kwargs['codec'] = codec output_kwargs['codec'] = _to_ffmpeg_codec(codec)
process = ( process = (
ffmpeg ffmpeg
.input('pipe:', format='f32le', **input_kwargs) .input('pipe:', format='f32le', **input_kwargs)