Evaluate speech enhancemnt model performance
Project description
Speech Enhancement Model Evaluation
This is a python package to evaluate your speech enhancement model performance.
Installation
pip install model-evaluation-777 -i https://pypi.org/simple
PESQ and STOI Evaluation
def evaluate(ref_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/clean/", deg_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/noisy/", fs=16000):
'''
Args:
ref_dir: reference signal directory
deg_dir: degration signal directory
fs: sample rate
Returns:
pesq_score_average, stoi_score_average
'''
from model_evaluation import stoi_pesq
pesq_score, stoi_score = stoi_pesq.evaluate(ref_dir=ref_dir, deg_dir=deg_dir, fs=16000)
Plot SNR vs PESQ
def plot_snr_pesq(out_dir, enh_dir, ref_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/clean/", deg_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/noisy/", fs=16000):
'''
Args:
out_dir: directory to save figure
enh_dir: enhanced signal directory
ref_dir: reference signal directory
deg_dir: degration signal directory
fs: sample rate
Returns:
snr vs pesq score data
'''
The SNR vs PESQ figure will save to out_dir automatically.
from model_evaluation import stoi_pesq
data = stoi_pesq.plot_snr_pesq(out_dir=out_dir, enh_dir=enh_dir)
Plot SNR vs STOI
def plot_snr_stoi(out_dir, enh_dir, ref_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/clean/", deg_dir="/media/youwei/Chauncey's/VERSO_Dataset/test/noisy/", fs=16000):
'''
Args:
out_dir: directory to save figure
enh_dir: enhanced signal directory
ref_dir: reference signal directory
deg_dir: degration signal directory
fs: sample rate
Returns:
snr vs stoi score data
'''
The SNR vs STOI figure will save to out_dir automatically.
from model_evaluation import stoi_pesq
data = stoi_pesq.plot_snr_stoi(out_dir=out_dir, enh_dir=enh_dir)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for model-evaluation-777-1.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 90c91e99841b9d102fda6fef14ba21af655b52f9a7f8c3f0db5a2c6157de56dd |
|
MD5 | 594093092c1633f0e4993ebc92914a20 |
|
BLAKE2b-256 | fb4abbb1ae3e06866039c11c427d37f54bd7502f7c412f0aa953ecb1c5024301 |
Close
Hashes for model_evaluation_777-1.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ded20cf0a413f77035140b1442929dbf044abdb8d8448af02dc57c7af8afc117 |
|
MD5 | 5b06665c667e8df0280532bf4b9c77aa |
|
BLAKE2b-256 | 08073bbba6b5777b8383f35d1504b548b1e07535e793276770eb704301aa9588 |