2015-11-09 22:31:39 +03:00
|
|
|
#ifndef REGION_LAYER_H
|
|
|
|
#define REGION_LAYER_H
|
2014-07-14 09:07:51 +04:00
|
|
|
|
2015-05-11 23:46:49 +03:00
|
|
|
#include "layer.h"
|
2015-12-14 22:57:10 +03:00
|
|
|
#include "network.h"
|
2015-03-12 08:20:15 +03:00
|
|
|
|
2015-05-11 23:46:49 +03:00
|
|
|
typedef layer detection_layer;
|
2014-07-14 09:07:51 +04:00
|
|
|
|
2015-11-09 22:31:39 +03:00
|
|
|
detection_layer make_detection_layer(int batch, int inputs, int n, int size, int classes, int coords, int rescore);
|
2015-05-11 23:46:49 +03:00
|
|
|
void forward_detection_layer(const detection_layer l, network_state state);
|
|
|
|
void backward_detection_layer(const detection_layer l, network_state state);
|
2015-03-05 01:56:38 +03:00
|
|
|
|
|
|
|
#ifdef GPU
|
2015-05-11 23:46:49 +03:00
|
|
|
void forward_detection_layer_gpu(const detection_layer l, network_state state);
|
|
|
|
void backward_detection_layer_gpu(detection_layer l, network_state state);
|
2015-03-05 01:56:38 +03:00
|
|
|
#endif
|
2014-07-14 09:07:51 +04:00
|
|
|
|
|
|
|
#endif
|