1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
| // Oh boy, why am I about to do this....
| #ifndef NETWORK_H
| #define NETWORK_H
|
| #include "image.h"
|
| typedef enum {
| CONVOLUTIONAL,
| CONNECTED,
| MAXPOOL
| } LAYER_TYPE;
|
| typedef struct {
| int n;
| void **layers;
| LAYER_TYPE *types;
| } network;
|
| network make_network(int n);
| void run_network(image input, network net);
| void learn_network(image input, network net);
| void update_network(network net, double step);
| double *get_network_output(network net);
| double *get_network_output_layer(network net, int i);
| int get_network_output_size_layer(network net, int i);
| image get_network_image(network net);
| image get_network_image_layer(network net, int i);
|
| #endif
|
|
