mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

lots of stuff

Browse Source
This commit is contained in:
Joseph Redmon
2016-01-28 12:30:38 -08:00
parent 1578ec70d7
commit 913d355ec1
35 changed files with 913 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/layer.h
View File

@ -21,11 +21,12 @@ typedef enum {
AVGPOOL,
LOCAL,
SHORTCUT,
ACTIVE
ACTIVE,
RNN
} LAYER_TYPE;
typedef enum{
SSE, MASKED
SSE, MASKED, SMOOTH
} COST_TYPE;
struct layer{
@ -50,6 +51,9 @@ struct layer{
int sqrt;
int flip;
int index;
int binary;
int steps;
int hidden;
float angle;
float jitter;
float saturation;
@ -77,6 +81,7 @@ struct layer{
int dontload;
int dontloadscales;
float temperature;
float probability;
float scale;
@ -85,6 +90,9 @@ struct layer{
float *cost;
float *filters;
float *filter_updates;
float *state;
float *binary_filters;
float *biases;
float *bias_updates;
@ -107,14 +115,28 @@ struct layer{
float * mean;
float * variance;
float * mean_delta;
float * variance_delta;
float * rolling_mean;
float * rolling_variance;
float * x;
float * x_norm;
struct layer *input_layer;
struct layer *self_layer;
struct layer *output_layer;
#ifdef GPU
int *indexes_gpu;
float * state_gpu;
float * filters_gpu;
float * filter_updates_gpu;
float *binary_filters_gpu;
float *mean_filters_gpu;
float * spatial_mean_gpu;
float * spatial_variance_gpu;