From 96b578db63e0cb5813afa17e31e0c15e1e90052c Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Mon, 11 May 2015 13:54:58 -0700 Subject: [PATCH] whoops! --- src/layer.h | 88 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 src/layer.h diff --git a/src/layer.h b/src/layer.h new file mode 100644 index 00000000..86e64a35 --- /dev/null +++ b/src/layer.h @@ -0,0 +1,88 @@ +#ifndef BASE_LAYER_H +#define BASE_LAYER_H + +#include "activations.h" + +typedef enum { + CONVOLUTIONAL, + DECONVOLUTIONAL, + CONNECTED, + MAXPOOL, + SOFTMAX, + DETECTION, + DROPOUT, + CROP, + ROUTE, + COST +} LAYER_TYPE; + +typedef enum{ + SSE, MASKED +} COST_TYPE; + +typedef struct { + LAYER_TYPE type; + ACTIVATION activation; + COST_TYPE cost_type; + int batch; + int inputs; + int outputs; + int h,w,c; + int out_h, out_w, out_c; + int n; + int groups; + int size; + int stride; + int pad; + int crop_width; + int crop_height; + int flip; + float angle; + float saturation; + float exposure; + int classes; + int coords; + int background; + int rescore; + int nuisance; + int does_cost; + float probability; + float scale; + int *indexes; + float *rand; + float *cost; + float *filters; + float *filter_updates; + + float *biases; + float *bias_updates; + + float *weights; + float *weight_updates; + + float *col_image; + int * input_layers; + int * input_sizes; + float * delta; + float * output; + + #ifdef GPU + int *indexes_gpu; + float * filters_gpu; + float * filter_updates_gpu; + + float * col_image_gpu; + + float * weights_gpu; + float * biases_gpu; + + float * weight_updates_gpu; + float * bias_updates_gpu; + + float * output_gpu; + float * delta_gpu; + float * rand_gpu; + #endif +} layer; + +#endif