mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-17 00:23:14 +00:00
Justin John
51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
|
|
#ifdef USE_TENSORFLOW1
|
|
#include "../Tensorflow/tensorflow.hpp"
|
|
#elif USE_TENSORFLOW2
|
|
#include "../Tensorflow2/tensorflow2.h"
|
|
#endif
|
|
|
|
struct neuralNetworkModel
|
|
{
|
|
#ifdef USE_TENSORFLOW1
|
|
struct TensorflowInstance model;
|
|
#elif USE_TENSORFLOW2
|
|
struct Tensorflow2Instance model;
|
|
#else
|
|
#warning "No tensorflow support signaled through a C define.."
|
|
void * model;
|
|
#endif
|
|
};
|
|
|
|
|
|
void neuralNetworkPrintVersion();
|
|
|
|
char * neuralNetworkGetPath(struct neuralNetworkModel * nn);
|
|
|
|
int neuralNetworkLoad(
|
|
struct neuralNetworkModel * nn,
|
|
const char * filename,
|
|
const char * inputTensor,
|
|
const char * outputTensor,
|
|
unsigned int numberOfInputElements,
|
|
unsigned int numberOfOutputElements,
|
|
unsigned int forceCPU
|
|
);
|
|
|
|
|
|
|
|
|
|
int neuralNetworkUnload(
|
|
struct neuralNetworkModel * nn
|
|
);
|
|
|
|
|
|
std::vector<float> neuralNetworkExecute(
|
|
struct neuralNetworkModel * nn,
|
|
std::vector<float> input
|
|
);
|