updating
This commit is contained in:
@@ -0,0 +1,26 @@
|
||||
#ifndef NEURALNETWORK_H
|
||||
#define NEURALNETWORK_H
|
||||
|
||||
#include "imageInput.h"
|
||||
#include "matrix.h"
|
||||
|
||||
typedef void (*ActivationFunctionType)(Matrix *X);
|
||||
|
||||
typedef struct
|
||||
{
|
||||
Matrix weights;
|
||||
Matrix biases;
|
||||
ActivationFunctionType activation;
|
||||
} Layer;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
Layer *layers;
|
||||
unsigned int numberOfLayers;
|
||||
} NeuralNetwork;
|
||||
|
||||
NeuralNetwork loadModel(const char *path);
|
||||
unsigned char *predict(const NeuralNetwork model, const GrayScaleImage images[], unsigned int numberOfImages);
|
||||
void clearModel(NeuralNetwork *model);
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user