Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
qq_36265860
spleeter
提交
c96897df
S
spleeter
项目概览
qq_36265860
/
spleeter
与 Fork 源项目一致
从无法访问的项目Fork
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
S
spleeter
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
c96897df
编写于
12月 08, 2020
作者:
F
Faylixe
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
🐛
fix backend resolution
上级
232bf0d3
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
15 addition
and
7 deletion
+15
-7
spleeter/audio/__init__.py
spleeter/audio/__init__.py
+14
-0
spleeter/separator.py
spleeter/separator.py
+1
-7
未找到文件。
spleeter/audio/__init__.py
浏览文件 @
c96897df
...
...
@@ -12,6 +12,11 @@
from
enum
import
Enum
# pyright: reportMissingImports=false
# pylint: disable=import-error
import
tensorflow
as
tf
# pylint: enable=import-error
__email__
=
'spleeter@deezer.com'
__author__
=
'Deezer Research'
__license__
=
'MIT License'
...
...
@@ -34,3 +39,12 @@ class STFTBackend(str, Enum):
AUTO
:
str
=
'auto'
TENSORFLOW
:
str
=
'tensorflow'
LIBROSA
:
str
=
'librosa'
def
resolve
(
cls
:
type
,
backend
:
str
)
->
str
:
if
backend
not
in
cls
.
__members__
.
items
():
raise
ValueError
(
f
'Unsupported backend
{
backend
}
'
)
if
backend
==
cls
.
AUTO
:
if
len
(
tf
.
config
.
list_physical_devices
(
'GPU'
)):
return
cls
.
TENSORFLOW
return
STFTBackend
.
LIBROSA
return
backend
spleeter/separator.py
浏览文件 @
c96897df
...
...
@@ -129,8 +129,7 @@ class Separator(object):
else
:
self
.
_pool
=
None
self
.
_tasks
=
[]
# NOTE: provide type check here ?
self
.
_params
[
'stft_backend'
]
=
stft_backend
self
.
_params
[
'stft_backend'
]
=
STFTBackend
.
resolve
(
stft_backend
)
self
.
_data_generator
=
DataGenerator
()
def
__del__
(
self
)
->
None
:
...
...
@@ -333,11 +332,6 @@ class Separator(object):
(Optional) string describing the waveform (e.g. filename).
"""
backend
:
str
=
self
.
_params
[
'stft_backend'
]
if
backend
==
STFTBackend
.
AUTO
:
if
len
(
tf
.
config
.
list_physical_devices
(
'GPU'
)):
backend
=
STFTBackend
.
TENSORFLOW
else
:
backend
=
STFTBackend
.
LIBROSA
if
backend
==
STFTBackend
.
TENSORFLOW
:
return
self
.
_separate_tensorflow
(
waveform
,
audio_descriptor
)
elif
backend
==
STFTBackend
.
LIBROSA
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录