@@ -5,59 +5,54 @@
|
||||
// This file shows how to use sherpa-onnx C API
|
||||
// to decode a file.
|
||||
|
||||
#include "cargs.h"
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "cargs.h"
|
||||
#include "sherpa-onnx/c-api/c-api.h"
|
||||
|
||||
static struct cag_option options[] = {
|
||||
{
|
||||
.identifier = 't',
|
||||
.access_letters = NULL,
|
||||
.access_name = "tokens",
|
||||
.value_name = "tokens",
|
||||
.description = "Tokens file"
|
||||
}, {
|
||||
.identifier = 'e',
|
||||
.access_letters = NULL,
|
||||
.access_name = "encoder",
|
||||
.value_name = "encoder",
|
||||
.description = "Encoder ONNX file"
|
||||
}, {
|
||||
.identifier = 'd',
|
||||
.access_letters = NULL,
|
||||
.access_name = "decoder",
|
||||
.value_name = "decoder",
|
||||
.description = "Decoder ONNX file"
|
||||
}, {
|
||||
.identifier = 'j',
|
||||
.access_letters = NULL,
|
||||
.access_name = "joiner",
|
||||
.value_name = "joiner",
|
||||
.description = "Joiner ONNX file"
|
||||
}, {
|
||||
.identifier = 'n',
|
||||
.access_letters = NULL,
|
||||
.access_name = "num-threads",
|
||||
.value_name = "num-threads",
|
||||
.description = "Number of threads"
|
||||
}, {
|
||||
.identifier = 'p',
|
||||
.access_letters = NULL,
|
||||
.access_name = "provider",
|
||||
.value_name = "provider",
|
||||
.description = "Provider: cpu (default), cuda, coreml"
|
||||
}, {
|
||||
.identifier = 'm',
|
||||
.access_letters = NULL,
|
||||
.access_name = "decoding-method",
|
||||
.value_name = "decoding-method",
|
||||
.description =
|
||||
"Decoding method: greedy_search (default), modified_beam_search"
|
||||
}
|
||||
};
|
||||
{.identifier = 'h',
|
||||
.access_letters = "h",
|
||||
.access_name = "help",
|
||||
.description = "Show help"},
|
||||
{.identifier = 't',
|
||||
.access_letters = NULL,
|
||||
.access_name = "tokens",
|
||||
.value_name = "tokens",
|
||||
.description = "Tokens file"},
|
||||
{.identifier = 'e',
|
||||
.access_letters = NULL,
|
||||
.access_name = "encoder",
|
||||
.value_name = "encoder",
|
||||
.description = "Encoder ONNX file"},
|
||||
{.identifier = 'd',
|
||||
.access_letters = NULL,
|
||||
.access_name = "decoder",
|
||||
.value_name = "decoder",
|
||||
.description = "Decoder ONNX file"},
|
||||
{.identifier = 'j',
|
||||
.access_letters = NULL,
|
||||
.access_name = "joiner",
|
||||
.value_name = "joiner",
|
||||
.description = "Joiner ONNX file"},
|
||||
{.identifier = 'n',
|
||||
.access_letters = NULL,
|
||||
.access_name = "num-threads",
|
||||
.value_name = "num-threads",
|
||||
.description = "Number of threads"},
|
||||
{.identifier = 'p',
|
||||
.access_letters = NULL,
|
||||
.access_name = "provider",
|
||||
.value_name = "provider",
|
||||
.description = "Provider: cpu (default), cuda, coreml"},
|
||||
{.identifier = 'm',
|
||||
.access_letters = NULL,
|
||||
.access_name = "decoding-method",
|
||||
.value_name = "decoding-method",
|
||||
.description =
|
||||
"Decoding method: greedy_search (default), modified_beam_search"}};
|
||||
|
||||
const char *kUsage =
|
||||
"\n"
|
||||
@@ -67,6 +62,7 @@ const char *kUsage =
|
||||
" --encoder=/path/to/encoder.onnx \\\n"
|
||||
" --decoder=/path/to/decoder.onnx \\\n"
|
||||
" --joiner=/path/to/joiner.onnx \\\n"
|
||||
" --provider=cpu \\\n"
|
||||
" /path/to/foo.wav\n"
|
||||
"\n\n"
|
||||
"Default num_threads is 1.\n"
|
||||
@@ -77,6 +73,11 @@ const char *kUsage =
|
||||
"for a list of pre-trained models to download.\n";
|
||||
|
||||
int32_t main(int32_t argc, char *argv[]) {
|
||||
if (argc < 6) {
|
||||
fprintf(stderr, "%s\n", kUsage);
|
||||
exit(0);
|
||||
}
|
||||
|
||||
SherpaOnnxOnlineRecognizerConfig config;
|
||||
|
||||
config.model_config.debug = 0;
|
||||
@@ -105,19 +106,38 @@ int32_t main(int32_t argc, char *argv[]) {
|
||||
identifier = cag_option_get(&context);
|
||||
value = cag_option_get_value(&context);
|
||||
switch (identifier) {
|
||||
case 't': config.model_config.tokens = value; break;
|
||||
case 'e': config.model_config.encoder = value; break;
|
||||
case 'd': config.model_config.decoder = value; break;
|
||||
case 'j': config.model_config.joiner = value; break;
|
||||
case 'n': config.model_config.num_threads = atoi(value); break;
|
||||
case 'p': config.model_config.provider = value; break;
|
||||
case 'm': config.decoding_method = value; break;
|
||||
default:
|
||||
case 't':
|
||||
config.model_config.tokens = value;
|
||||
break;
|
||||
case 'e':
|
||||
config.model_config.encoder = value;
|
||||
break;
|
||||
case 'd':
|
||||
config.model_config.decoder = value;
|
||||
break;
|
||||
case 'j':
|
||||
config.model_config.joiner = value;
|
||||
break;
|
||||
case 'n':
|
||||
config.model_config.num_threads = atoi(value);
|
||||
break;
|
||||
case 'p':
|
||||
config.model_config.provider = value;
|
||||
break;
|
||||
case 'm':
|
||||
config.decoding_method = value;
|
||||
break;
|
||||
case 'h': {
|
||||
fprintf(stderr, "%s\n", kUsage);
|
||||
exit(0);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
// do nothing as config already have valid default values
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
SherpaOnnxOnlineRecognizer *recognizer = CreateOnlineRecognizer(&config);
|
||||
SherpaOnnxOnlineStream *stream = CreateOnlineStream(recognizer);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user