Joseph Redmon
2015-08-25 9d42f49a240136a8cd643cdc1f98230d4f22b05e
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#ifndef REGION_LAYER_H
#define REGION_LAYER_H
 
#include "params.h"
#include "layer.h"
 
typedef layer region_layer;
 
region_layer make_region_layer(int batch, int inputs, int n, int classes, int coords, int rescore);
void forward_region_layer(const region_layer l, network_state state);
void backward_region_layer(const region_layer l, network_state state);
 
#ifdef GPU
void forward_region_layer_gpu(const region_layer l, network_state state);
void backward_region_layer_gpu(region_layer l, network_state state);
#endif
 
#endif