forked from Qortal/Brooklyn
32 lines
668 B
C++
32 lines
668 B
C++
|
//
|
||
|
|
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
//
|
||
|
|
# pragma once
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <map>
|
||
|
|
#include "ArmnnNetworkExecutor.hpp"
|
||
|
|
|
||
|
|
namespace kws
|
||
|
|
{
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @brief Decodes quantised last layer of model output
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
class Decoder
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
int quantisationOffset;
|
||
|
|
float quantisationScale;
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
Decoder(int quantisationOffset, float quantisationScale) : quantisationOffset(quantisationOffset),
|
||
|
|
quantisationScale(quantisationScale) {}
|
||
|
|
|
||
|
|
std::pair<int, float> decodeOutput(std::vector<int8_t>& modelOutput);
|
||
|
|
|
||
|
|
};
|
||
|
|
} // namespace kws
|