Joseph Redmon
2013-11-04 41bcfac86fa4dc856f3d1894efeee0f55f86f7b0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#ifndef CONVOLUTIONAL_LAYER_H
#define CONVOLUTIONAL_LAYER_H
 
#include "image.h"
 
typedef struct {
    int n;
    int stride;
    image *kernels;
    image *kernel_updates;
    image upsampled;
    image output;
} convolutional_layer;
 
convolutional_layer make_convolutional_layer(int w, int h, int c, int n, int size, int stride);
void run_convolutional_layer(const image input, const convolutional_layer layer);
void backpropagate_layer(image input, convolutional_layer layer);
void backpropagate_layer_convolve(image input, convolutional_layer layer);
 
#endif