38c072dcb2
* add export of per-token scores (ys, lm, context) - for best path of the modified-beam-search decoding of transducer * refactoring JSON export of OnlineRecognitionResult, extending pybind11 API of OnlineRecognitionResult * export per-token scores also for greedy-search (online-transducer) - export un-scaled lm_probs (modified-beam search, online-transducer) - polishing * fill lm_probs/context_scores only if LM/ContextGraph is present (make Result smaller)
75 lines
1.8 KiB
C++
75 lines
1.8 KiB
C++
// sherpa-onnx/csrc/online-transducer-decoder.cc
|
|
//
|
|
// Copyright (c) 2023 Xiaomi Corporation
|
|
|
|
#include "sherpa-onnx/csrc/online-transducer-decoder.h"
|
|
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "onnxruntime_cxx_api.h" // NOLINT
|
|
#include "sherpa-onnx/csrc/onnx-utils.h"
|
|
|
|
namespace sherpa_onnx {
|
|
|
|
OnlineTransducerDecoderResult::OnlineTransducerDecoderResult(
|
|
const OnlineTransducerDecoderResult &other)
|
|
: OnlineTransducerDecoderResult() {
|
|
*this = other;
|
|
}
|
|
|
|
OnlineTransducerDecoderResult &OnlineTransducerDecoderResult::operator=(
|
|
const OnlineTransducerDecoderResult &other) {
|
|
if (this == &other) {
|
|
return *this;
|
|
}
|
|
|
|
tokens = other.tokens;
|
|
num_trailing_blanks = other.num_trailing_blanks;
|
|
|
|
Ort::AllocatorWithDefaultOptions allocator;
|
|
if (other.decoder_out) {
|
|
decoder_out = Clone(allocator, &other.decoder_out);
|
|
}
|
|
|
|
hyps = other.hyps;
|
|
|
|
frame_offset = other.frame_offset;
|
|
timestamps = other.timestamps;
|
|
|
|
ys_probs = other.ys_probs;
|
|
lm_probs = other.lm_probs;
|
|
context_scores = other.context_scores;
|
|
|
|
return *this;
|
|
}
|
|
|
|
OnlineTransducerDecoderResult::OnlineTransducerDecoderResult(
|
|
OnlineTransducerDecoderResult &&other)
|
|
: OnlineTransducerDecoderResult() {
|
|
*this = std::move(other);
|
|
}
|
|
|
|
OnlineTransducerDecoderResult &OnlineTransducerDecoderResult::operator=(
|
|
OnlineTransducerDecoderResult &&other) {
|
|
if (this == &other) {
|
|
return *this;
|
|
}
|
|
|
|
tokens = std::move(other.tokens);
|
|
num_trailing_blanks = other.num_trailing_blanks;
|
|
decoder_out = std::move(other.decoder_out);
|
|
hyps = std::move(other.hyps);
|
|
|
|
frame_offset = other.frame_offset;
|
|
timestamps = std::move(other.timestamps);
|
|
|
|
ys_probs = std::move(other.ys_probs);
|
|
lm_probs = std::move(other.lm_probs);
|
|
context_scores = std::move(other.context_scores);
|
|
|
|
return *this;
|
|
}
|
|
|
|
} // namespace sherpa_onnx
|