lyblsgo
2023-03-16 76164048454ef781c29be941d6cc3a2801a349b1
fix c++ grpc onnx server
4个文件已修改
60 ■■■■■ 已修改文件
funasr/runtime/grpc/Readme.md 9 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/grpc/paraformer_server.cc 36 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/grpc/paraformer_server.h 11 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/grpc/rebuild.sh 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/grpc/Readme.md
@@ -27,6 +27,7 @@
      -DCMAKE_INSTALL_PREFIX=$GRPC_INSTALL_DIR \
      ../..
make
make install
popd
echo "export GRPC_INSTALL_DIR=/data/soft/grpc" >> ~/.bashrc
@@ -40,9 +41,15 @@
./rebuild.sh
```
#### Step 4. Start grpc paraformer server
```
Usage: ./cmake/build/paraformer_server port thread_num /path/to/model_file
./cmake/build/paraformer_server 10108 4 /data/asrmodel
```
#### Step 4. Start grpc python paraformer client  on PC with MIC
#### Step 5. Start grpc python paraformer client  on PC with MIC
```
cd ../python/grpc
python grpc_main_client_mic.py  --host $server_ip --port 10108
funasr/runtime/grpc/paraformer_server.cc
@@ -29,7 +29,8 @@
using paraformer::Response;
using paraformer::ASR;
ASRServicer::ASRServicer() {
ASRServicer::ASRServicer(const char* model_path, int thread_num) {
    AsrHanlde=RapidAsrInit(model_path, thread_num);
    std::cout << "ASRServicer init" << std::endl;
    init_flag = 0;
}
@@ -135,7 +136,7 @@
                }
                else {
                    RPASR_RESULT Result= RapidAsrRecogPCMBuffer(AsrHanlde, tmp_data.c_str(), data_len_int, RASR_NONE, NULL);   
                    std::string asr_result = "你好你好,我是asr识别结果。static";
                    std::string asr_result = ((RPASR_RECOG_RESULT*)Result)->msg;
                    auto end_time = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count();
                    std::string delay_str = std::to_string(end_time - begin_time);
@@ -169,19 +170,26 @@
}
void RunServer() {
  std::string server_address("0.0.0.0:10108");
  ASRServicer service;
void RunServer(const std::string& port, int thread_num, const char* model_path) {
    std::string server_address;
    server_address = "0.0.0.0:" + port;
    ASRServicer service(model_path, thread_num);
  ServerBuilder builder;
  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
  builder.RegisterService(&service);
  std::unique_ptr<Server> server(builder.BuildAndStart());
  std::cout << "Server listening on " << server_address << std::endl;
  server->Wait();
    ServerBuilder builder;
    builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
    builder.RegisterService(&service);
    std::unique_ptr<Server> server(builder.BuildAndStart());
    std::cout << "Server listening on " << server_address << std::endl;
    server->Wait();
}
int main(int argc, char** argv) {
 RunServer();
 return 0;
int main(int argc, char* argv[]) {
    if (argc < 3)
    {
        printf("Usage: %s port thread_num /path/to/model_file\n", argv[0]);
        exit(-1);
    }
    RunServer(argv[1], atoi(argv[2]), argv[3]);
    return 0;
}
funasr/runtime/grpc/paraformer_server.h
@@ -31,6 +31,12 @@
using paraformer::Response;
using paraformer::ASR;
typedef struct
{
    std::string msg;
    float  snippet_time;
}RPASR_RECOG_RESULT;
class ASRServicer final : public ASR::Service {
  private:
@@ -39,13 +45,12 @@
    std::unordered_map<std::string, std::string> client_transcription;
  public:
    ASRServicer();
    ASRServicer(const char* model_path, int thread_num);
    void clear_states(const std::string& user);
    void clear_buffers(const std::string& user);
    void clear_transcriptions(const std::string& user);
    void disconnect(const std::string& user);
    grpc::Status Recognize(grpc::ServerContext* context, grpc::ServerReaderWriter<Response, Request>* stream);
    int nThreadNum = 4;
    RPASR_HANDLE AsrHanlde=RapidAsrInit("/data/asrmodel/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/", nThreadNum);
    RPASR_HANDLE AsrHanlde;
    
};
funasr/runtime/grpc/rebuild.sh
@@ -10,7 +10,3 @@
echo "Build cmake/build/paraformer_server successfully!"
echo "Start server"
./paraformer_server