File size: 434 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 |
#pragma once
#include "ext/CppFlow/include/Model.h"
#include "VoxCommon.hpp"
class MultiBandMelGAN
{
private:
Model* MelGAN;
public:
bool Initialize(const std::string& VocoderPath);
// Do MultiBand MelGAN inference including PQMF
// -> InMel: Mel spectrogram (shape [1, xx, 80])
// <- Returns: Tensor data [4, xx, 1]
TFTensor<float> DoInference(const TFTensor<float>& InMel);
MultiBandMelGAN();
~MultiBandMelGAN();
};
|