Go to the documentation of this file.00001 #ifndef REGION_LAYER_H
00002 #define REGION_LAYER_H
00003
00004 #include "layer.h"
00005 #include "network.h"
00006
00007 typedef layer region_layer;
00008
00009 region_layer make_region_layer(int batch, int h, int w, int n, int classes, int coords);
00010 void forward_region_layer(const region_layer l, network_state state);
00011 void backward_region_layer(const region_layer l, network_state state);
00012 void get_region_boxes(layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness);
00013 void resize_region_layer(layer *l, int w, int h);
00014
00015 #ifdef GPU
00016 void forward_region_layer_gpu(const region_layer l, network_state state);
00017 void backward_region_layer_gpu(region_layer l, network_state state);
00018 #endif
00019
00020 #endif