File size: 698 Bytes
d5ee97c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
#ifndef MELGENERATETF_H
#define MELGENERATETF_H
#include "TfliteBase.h"
class MelGenerateTF : public TfliteBase
{
public:
MelGenerateTF(const char* modelFilename):TfliteBase(modelFilename),
inputIndexs(interpreter->inputs()),
ouptIndex(interpreter->outputs()[1]) {};
MelGenData infer(const std::vector<int32_t> inputIds);
private:
std::vector<int32_t> _speakerId{0};
std::vector<float> _speedRatio{1.0};
std::vector<float> _f0Ratio{1.0};
std::vector<float> _enegyRatio{1.0};
const std::vector<int32_t> inputIndexs;
const int32_t ouptIndex;
};
#endif // MELGENERATETF_H |