Initial commit: FunASR Speech Recognition Toolkit
Update API Documentation / build-api-docs (push) Has been cancelled
Update API Documentation / build-api-docs (push) Has been cancelled
Add complete FunASR codebase including models, runtime, and documentation.
This commit is contained in:
@@ -0,0 +1,68 @@
|
||||
/**
|
||||
* Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights
|
||||
* Reserved. MIT License (https://opensource.org/licenses/MIT)
|
||||
*/
|
||||
/* 2023 by burkliu(刘柏基) liubaiji@xverse.cn */
|
||||
|
||||
#include <string>
|
||||
#include <thread>
|
||||
#include <mutex>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "grpcpp/server_builder.h"
|
||||
#include "paraformer.grpc.pb.h"
|
||||
#include "funasrruntime.h"
|
||||
#include "tclap/CmdLine.h"
|
||||
#include "com-define.h"
|
||||
#include "glog/logging.h"
|
||||
|
||||
using paraformer::WavFormat;
|
||||
using paraformer::DecodeMode;
|
||||
using paraformer::Request;
|
||||
using paraformer::Response;
|
||||
using paraformer::ASR;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
std::string msg;
|
||||
float snippet_time;
|
||||
} FUNASR_RECOG_RESULT;
|
||||
|
||||
class GrpcEngine {
|
||||
public:
|
||||
GrpcEngine(grpc::ServerReaderWriter<Response, Request>* stream, std::shared_ptr<FUNASR_HANDLE> asr_handler);
|
||||
void operator()();
|
||||
|
||||
private:
|
||||
void DecodeThreadFunc();
|
||||
void OnSpeechStart();
|
||||
void OnSpeechData();
|
||||
void OnSpeechEnd();
|
||||
|
||||
grpc::ServerReaderWriter<Response, Request>* stream_;
|
||||
std::shared_ptr<Request> request_;
|
||||
std::shared_ptr<Response> response_;
|
||||
std::shared_ptr<FUNASR_HANDLE> asr_handler_;
|
||||
std::string audio_buffer_;
|
||||
std::shared_ptr<std::thread> decode_thread_ = nullptr;
|
||||
bool is_start_ = false;
|
||||
bool is_end_ = false;
|
||||
|
||||
std::vector<int> chunk_size_ = {5, 10, 5};
|
||||
int sampling_rate_ = 16000;
|
||||
std::string encoding_;
|
||||
ASR_TYPE mode_ = ASR_TWO_PASS;
|
||||
int step_duration_ms_ = 100;
|
||||
|
||||
std::unique_ptr<std::mutex> p_mutex_= std::make_unique<std::mutex>(); // mutex is not moveable
|
||||
};
|
||||
|
||||
class GrpcService final : public ASR::Service {
|
||||
public:
|
||||
GrpcService(std::map<std::string, std::string>& config, int num_thread);
|
||||
grpc::Status Recognize(grpc::ServerContext* context, grpc::ServerReaderWriter<Response, Request>* stream);
|
||||
|
||||
private:
|
||||
std::map<std::string, std::string> config_;
|
||||
std::shared_ptr<FUNASR_HANDLE> asr_handler_;
|
||||
};
|
||||
Reference in New Issue
Block a user