97e349ace7
This reverts commit 2491cb73820fe82923b848dfcab6772b4b0addb0. Reason for revert: broke internal build Original change's description: > AGC2 RNN VAD: Recurrent Neural Network impl > > RNN implementation for the AGC2 VAD that includes a fully connected > layer and a gated recurrent unit layer. > > Bug: webrtc:9076 > Change-Id: Ibb8b0b4e9213f09eb9dbe118bbdc94d7e8e4f91b > Reviewed-on: https://webrtc-review.googlesource.com/72060 > Reviewed-by: Patrik Höglund <phoglund@webrtc.org> > Reviewed-by: Alex Loiko <aleloi@webrtc.org> > Reviewed-by: Ivo Creusen <ivoc@webrtc.org> > Commit-Queue: Alessio Bazzica <alessiob@webrtc.org> > Cr-Commit-Position: refs/heads/master@{#23101} TBR=phoglund@webrtc.org,alessiob@webrtc.org,aleloi@webrtc.org,ivoc@webrtc.org Change-Id: Ic311c4b7d79094e959d3a2c4a53c398f34c954e2 No-Presubmit: true No-Tree-Checks: true No-Try: true Bug: webrtc:9076 Reviewed-on: https://webrtc-review.googlesource.com/74200 Reviewed-by: Sam Zackrisson <saza@webrtc.org> Commit-Queue: Sam Zackrisson <saza@webrtc.org> Cr-Commit-Position: refs/heads/master@{#23103}
59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
/*
|
|
* Copyright (c) 2018 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/agc2/rnn_vad/test_utils.h"
|
|
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/ptr_util.h"
|
|
#include "test/gtest.h"
|
|
#include "test/testsupport/fileutils.h"
|
|
|
|
namespace webrtc {
|
|
namespace rnn_vad {
|
|
namespace test {
|
|
namespace {
|
|
|
|
using ReaderPairType =
|
|
std::pair<std::unique_ptr<BinaryFileReader<float>>, const size_t>;
|
|
|
|
} // namespace
|
|
|
|
using webrtc::test::ResourcePath;
|
|
|
|
void ExpectNearAbsolute(rtc::ArrayView<const float> expected,
|
|
rtc::ArrayView<const float> computed,
|
|
float tolerance) {
|
|
ASSERT_EQ(expected.size(), computed.size());
|
|
for (size_t i = 0; i < expected.size(); ++i) {
|
|
SCOPED_TRACE(i);
|
|
EXPECT_NEAR(expected[i], computed[i], tolerance);
|
|
}
|
|
}
|
|
|
|
ReaderPairType CreatePitchBuffer24kHzReader() {
|
|
auto ptr = rtc::MakeUnique<BinaryFileReader<float>>(
|
|
ResourcePath("audio_processing/agc2/rnn_vad/pitch_buf_24k", "dat"), 864);
|
|
return {std::move(ptr),
|
|
rtc::CheckedDivExact(ptr->data_length(), static_cast<size_t>(864))};
|
|
}
|
|
|
|
ReaderPairType CreateLpResidualAndPitchPeriodGainReader() {
|
|
constexpr size_t num_lp_residual_coeffs = 864;
|
|
auto ptr = rtc::MakeUnique<BinaryFileReader<float>>(
|
|
ResourcePath("audio_processing/agc2/rnn_vad/pitch_lp_res", "dat"),
|
|
num_lp_residual_coeffs);
|
|
return {std::move(ptr),
|
|
rtc::CheckedDivExact(ptr->data_length(), 2 + num_lp_residual_coeffs)};
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace rnn_vad
|
|
} // namespace webrtc
|