From 96b578db63e0cb5813afa17e31e0c15e1e90052c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 11 May 2015 20:54:58 +0000
Subject: [PATCH] whoops!
---
src/layer.h | 88 ++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 88 insertions(+), 0 deletions(-)
diff --git a/src/layer.h b/src/layer.h
new file mode 100644
index 0000000..86e64a3
--- /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
--
Gitblit v1.10.0