Hui Zhang
3 years ago
parent
52477a1030
commit
357a364815
@ -0,0 +1,40 @@
|
||||
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "base/kaldi-common.h"
|
||||
#include "feat/feature-fbank.h"
|
||||
|
||||
#include "paddlespeech/audio/src/pybind/kaldi/feature_common.h"
|
||||
|
||||
namespace paddleaudio {
|
||||
namespace kaldi {
|
||||
|
||||
typedef StreamingFeatureTpl<::kaldi::FbankComputer> Fbank;
|
||||
|
||||
class KaldiFeatureWrapper {
|
||||
public:
|
||||
static KaldiFeatureWrapper* GetInstance();
|
||||
bool InitFbank(::kaldi::FbankOptions opts);
|
||||
py::array_t<double> ComputeFbank(const py::array_t<double> wav);
|
||||
int Dim() { return fbank_->Dim(); }
|
||||
void ResetFbank() { fbank_->Reset(); }
|
||||
|
||||
private:
|
||||
std::unique_ptr<paddleaudio::kaldi::Fbank> fbank_;
|
||||
};
|
||||
|
||||
} // namespace kaldi
|
||||
} // namespace paddleaudio
|
||||
@ -1,13 +0,0 @@
|
||||
|
||||
include_directories(
|
||||
${CMAKE_CURRENT_SOURCE_DIR}
|
||||
)
|
||||
|
||||
add_library(kaldi_feature
|
||||
kaldi_feature.cc
|
||||
kaldi_feature_wrapper.cc
|
||||
)
|
||||
target_link_libraries(kaldi_feature kaldi-fbank)
|
||||
|
||||
pybind11_add_module(kaldi_frontend kaldi_feature.cc kaldi_feature_wrapper.cc)
|
||||
target_link_libraries(kaldi_frontend PRIVATE kaldi_feature)
|
||||
@ -1,24 +0,0 @@
|
||||
|
||||
#include "base/kaldi-common.h"
|
||||
#include "feat/feature-fbank.h"
|
||||
#include "feature_common.h"
|
||||
|
||||
#pragma once
|
||||
|
||||
namespace paddleaudio {
|
||||
|
||||
typedef StreamingFeatureTpl<kaldi::FbankComputer> Fbank;
|
||||
|
||||
class KaldiFeatureWrapper {
|
||||
public:
|
||||
static KaldiFeatureWrapper* GetInstance();
|
||||
bool InitFbank(kaldi::FbankOptions opts);
|
||||
py::array_t<double> ComputeFbank(const py::array_t<double> wav);
|
||||
int Dim() { return fbank_->Dim(); }
|
||||
void ResetFbank() { fbank_->Reset(); }
|
||||
|
||||
private:
|
||||
std::unique_ptr<paddleaudio::Fbank> fbank_;
|
||||
};
|
||||
|
||||
} // namespace paddleaudio
|
||||
Loading…
Reference in new issue