lots of stuff

Joseph Redmon

2015-09-05 b5936b499abc94c0efffbcc99b5698574b59d860

 src/layer.h

@@ -13,7 +13,10 @@
    DROPOUT,
    CROP,
    ROUTE,
    COST
    COST,
    NORMALIZATION,
    REGION,
    AVGPOOL
} LAYER_TYPE;

typedef enum{
@@ -32,6 +35,7 @@
    int n;
    int groups;
    int size;
    int side;
    int stride;
    int pad;
    int crop_width;
@@ -44,10 +48,20 @@
    int coords;
    int background;
    int rescore;
    int nuisance;
    int objectness;
    int does_cost;
    int joint;
    int noadjust;

    float alpha;
    float beta;
    float kappa;

    int dontload;

    float probability;
    float scale;

    int *indexes;
    float *rand;
    float *cost;
@@ -65,6 +79,8 @@
    int   * input_sizes;
    float * delta;
    float * output;
    float * squared;
    float * norms;

    #ifdef GPU
    int *indexes_gpu;
@@ -82,7 +98,11 @@
    float * output_gpu;
    float * delta_gpu;
    float * rand_gpu;
    float * squared_gpu;
    float * norms_gpu;
    #endif
} layer;

void free_layer(layer);

#endif