Go to the documentation of this file.00001 #ifndef DETECTION_LAYER_H
00002 #define DETECTION_LAYER_H
00003
00004 #include "layer.h"
00005 #include "network.h"
00006
00007 typedef layer detection_layer;
00008
00009 detection_layer make_detection_layer(int batch, int inputs, int n, int size, int classes, int coords, int rescore);
00010 void forward_detection_layer(const detection_layer l, network_state state);
00011 void backward_detection_layer(const detection_layer l, network_state state);
00012 void get_detection_boxes(layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness);
00013
00014 #ifdef GPU
00015 void forward_detection_layer_gpu(const detection_layer l, network_state state);
00016 void backward_detection_layer_gpu(detection_layer l, network_state state);
00017 #endif
00018
00019 #endif