Joseph Redmon
2015-09-23 f996bd59a61338d8d51e2b19482d684f6dd04d0f
src/layer.h
@@ -13,7 +13,10 @@
    DROPOUT,
    CROP,
    ROUTE,
    COST
    COST,
    NORMALIZATION,
    REGION,
    AVGPOOL
} LAYER_TYPE;
typedef enum{
@@ -27,19 +30,23 @@
    int batch;
    int inputs;
    int outputs;
    int truths;
    int h,w,c;
    int out_h, out_w, out_c;
    int n;
    int groups;
    int size;
    int side;
    int stride;
    int pad;
    int crop_width;
    int crop_height;
    int sqrt;
    int flip;
    float angle;
    float saturation;
    float exposure;
    int softmax;
    int classes;
    int coords;
    int background;
@@ -47,9 +54,22 @@
    int objectness;
    int does_cost;
    int joint;
    int noadjust;
    float alpha;
    float beta;
    float kappa;
    float coord_scale;
    float object_scale;
    float noobject_scale;
    float class_scale;
    int dontload;
    float probability;
    float scale;
    int *indexes;
    float *rand;
    float *cost;
@@ -67,6 +87,8 @@
    int   * input_sizes;
    float * delta;
    float * output;
    float * squared;
    float * norms;
    #ifdef GPU
    int *indexes_gpu;
@@ -84,7 +106,11 @@
    float * output_gpu;
    float * delta_gpu;
    float * rand_gpu;
    float * squared_gpu;
    float * norms_gpu;
    #endif
} layer;
void free_layer(layer);
#endif