envelopecnn.h

#pragma once

#include <stdlib.h>
#include <math.h>
#include <tensorflow/cc/saved_model/loader.h>
#include <tensorflow/core/protobuf/saved_object_graph.pb.h>
#include <tensorflow/core/public/session.h>
#include <tensorflow/core/public/session_options.h>
#include <tensorflow/cc/saved_model/tag_constants.h>


class EnvelopeCNN {
private:
    typedef std::vector<std::pair<std::string, tensorflow::Tensor>> tensor_dict;
    tensorflow::Status status;
    tensorflow::Session *session;
    tensorflow::SavedModelBundle model;
    tensorflow::GraphDef graph_def;

public:
    EnvelopeCNN(std::string modelpath, std::string graphpath);
    void predict(float**** inp);
    ~EnvelopeCNN();

};