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: add musdb and museval check

Browse Source
This commit is contained in:
Félix Voituret
2019-11-07 00:04:18 +01:00
parent 30cd7c6435
commit f1f3efbe9c
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
spleeter/commands/evaluate.py
View File

@@ -13,6 +13,7 @@
--mus_dir /path/to/musdb dataset --mus_dir /path/to/musdb dataset
""" """
import sys
import json import json
from argparse import Namespace from argparse import Namespace
@@ -21,8 +22,6 @@ from glob import glob
from os.path import join, exists from os.path import join, exists
# pylint: disable=import-error # pylint: disable=import-error
import musdb
import museval
import numpy as np import numpy as np
import pandas as pd import pandas as pd
# pylint: enable=import-error # pylint: enable=import-error
@@ -30,6 +29,15 @@ import pandas as pd
from .separate import entrypoint as separate_entrypoint from .separate import entrypoint as separate_entrypoint
from ..utils.logging import get_logger from ..utils.logging import get_logger
try:
import musdb
import museval
except ImportError:
logger = get_logger()
logger.error('Extra dependencies musdb and museval not found')
logger.error('Please install musdb and museval first, abort')
sys.exit(1)
__email__ = 'research@deezer.com' __email__ = 'research@deezer.com'
__author__ = 'Deezer Research' __author__ = 'Deezer Research'
__license__ = 'MIT License' __license__ = 'MIT License'

7
spleeter/model/functions/unet.py
View File

@@ -3,10 +3,9 @@
""" """
This module contains building functions for U-net source separation source This module contains building functions for U-net source separation source
separation models. separation models. Each instrument is modeled by a single U-netconvolutional
Each instrument is modeled by a single U-net convolutional/deconvolutional / deconvolutional network that take a mix spectrogram as input and the
network that take a mix spectrogram as input and the estimated sound spectrogram estimated sound spectrogram as output.
as output.
""" """
from functools import partial from functools import partial