some more stuff

This commit is contained in:
Joseph Redmon
2016-08-05 15:27:07 -07:00
parent 9361292c42
commit 845ab75796
26 changed files with 1589 additions and 156 deletions

18
src/region_layer.h Normal file
View File

@@ -0,0 +1,18 @@
#ifndef REGION_LAYER_H
#define REGION_LAYER_H
#include "layer.h"
#include "network.h"
typedef layer region_layer;
region_layer make_region_layer(int batch, int h, int w, int n, int classes, int coords);
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