tipe/src/cnn/creation.h

#include "struct.h"

#ifndef DEF_CREATION_H
#define DEF_CREATION_H

/*
* Créé un réseau qui peut contenir max_size couche (dont celle d'input et d'output)
*/
Network* create_network(int max_size, int dropout, int initialisation, int input_dim, int input_depth);

/*
* Renvoie un réseau suivant l'architecture LeNet5
*/
Network* create_network_lenet5(int dropout, int activation, int initialisation);

/*
* Créé et alloue de la mémoire à une couche de type input cube
*/
void create_a_cube_input_layer(Network* network, int pos, int depth, int dim);

/*
* Créé et alloue de la mémoire à une couche de type ligne
*/
void create_a_line_input_layer(Network* network, int pos, int dim);

/*
* Ajoute au réseau une couche d'average pooling valide de dimension dim*dim
*/
void add_average_pooling(Network* network, int kernel_size, int activation);

/*
* Ajoute au réseau une couche d'average pooling valide de dimension dim*dim qui aplatit
*/
void add_average_pooling_flatten(Network* network, int kernel_size, int activation);

/*
* Ajoute une couche de convolution dim*dim au réseau et initialise les kernels
*/
void add_convolution(Network* network, int nb_filter, int kernel_size, int activation);

/*
* Ajoute une couche dense au réseau et initialise les poids et les biais
*/
void add_dense(Network* network, int input_units, int output_units, int activation);

#endif