You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.2 KiB
66 lines
2.2 KiB
/*
|
|
* Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "modules/audio_processing/vad/gmm.h"
|
|
|
|
#include <math.h>
|
|
|
|
#include "modules/audio_processing/vad/noise_gmm_tables.h"
|
|
#include "modules/audio_processing/vad/voice_gmm_tables.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
TEST(GmmTest, EvaluateGmm) {
|
|
GmmParameters noise_gmm;
|
|
GmmParameters voice_gmm;
|
|
|
|
// Setup noise GMM.
|
|
noise_gmm.dimension = kNoiseGmmDim;
|
|
noise_gmm.num_mixtures = kNoiseGmmNumMixtures;
|
|
noise_gmm.weight = kNoiseGmmWeights;
|
|
noise_gmm.mean = &kNoiseGmmMean[0][0];
|
|
noise_gmm.covar_inverse = &kNoiseGmmCovarInverse[0][0][0];
|
|
|
|
// Setup voice GMM.
|
|
voice_gmm.dimension = kVoiceGmmDim;
|
|
voice_gmm.num_mixtures = kVoiceGmmNumMixtures;
|
|
voice_gmm.weight = kVoiceGmmWeights;
|
|
voice_gmm.mean = &kVoiceGmmMean[0][0];
|
|
voice_gmm.covar_inverse = &kVoiceGmmCovarInverse[0][0][0];
|
|
|
|
// Test vectors. These are the mean of the GMM means.
|
|
const double kXVoice[kVoiceGmmDim] = {-1.35893162459863, 602.862491970368,
|
|
178.022069191324};
|
|
const double kXNoise[kNoiseGmmDim] = {-2.33443722724409, 2827.97828765184,
|
|
141.114178166812};
|
|
|
|
// Expected pdf values. These values are computed in MATLAB using EvalGmm.m
|
|
const double kPdfNoise = 1.88904409403101e-07;
|
|
const double kPdfVoice = 1.30453996982266e-06;
|
|
|
|
// Relative error should be smaller that the following value.
|
|
const double kAcceptedRelativeErr = 1e-10;
|
|
|
|
// Test Voice.
|
|
double pdf = EvaluateGmm(kXVoice, voice_gmm);
|
|
EXPECT_GT(pdf, 0);
|
|
double relative_error = fabs(pdf - kPdfVoice) / kPdfVoice;
|
|
EXPECT_LE(relative_error, kAcceptedRelativeErr);
|
|
|
|
// Test Noise.
|
|
pdf = EvaluateGmm(kXNoise, noise_gmm);
|
|
EXPECT_GT(pdf, 0);
|
|
relative_error = fabs(pdf - kPdfNoise) / kPdfNoise;
|
|
EXPECT_LE(relative_error, kAcceptedRelativeErr);
|
|
}
|
|
|
|
} // namespace webrtc
|