New YOLO

Joseph Redmon

2015-11-09 8c5364f58569eaeb5582a4915b36b24fc5570c76

 src/layer.h

@@ -15,7 +15,6 @@
    ROUTE,
    COST,
    NORMALIZATION,
    REGION,
    AVGPOOL
} LAYER_TYPE;

@@ -27,11 +26,9 @@
    LAYER_TYPE type;
    ACTIVATION activation;
    COST_TYPE cost_type;
    int batch_normalize;
    int batch;
    int forced;
    int object_logistic;
    int class_logistic;
    int coord_logistic;
    int inputs;
    int outputs;
    int truths;
@@ -51,6 +48,7 @@
    float jitter;
    float saturation;
    float exposure;
    float shift;
    int softmax;
    int classes;
    int coords;
@@ -71,6 +69,7 @@
    float class_scale;

    int dontload;
    int dontloadscales;

    float probability;
    float scale;
@@ -84,6 +83,9 @@
    float *biases;
    float *bias_updates;

    float *scales;
    float *scale_updates;

    float *weights;
    float *weight_updates;

@@ -95,18 +97,44 @@
    float * squared;
    float * norms;

    float * spatial_mean;
    float * mean;
    float * variance;

    float * rolling_mean;
    float * rolling_variance;

    #ifdef GPU
    int *indexes_gpu;
    float * filters_gpu;
    float * filter_updates_gpu;

    float * spatial_mean_gpu;
    float * spatial_variance_gpu;

    float * mean_gpu;
    float * variance_gpu;

    float * rolling_mean_gpu;
    float * rolling_variance_gpu;

    float * spatial_mean_delta_gpu;
    float * spatial_variance_delta_gpu;

    float * variance_delta_gpu;
    float * mean_delta_gpu;

    float * col_image_gpu;

    float * x_gpu;
    float * x_norm_gpu;
    float * weights_gpu;
    float * biases_gpu;
    float * scales_gpu;

    float * weight_updates_gpu;
    float * bias_updates_gpu;
    float * scale_updates_gpu;

    float * output_gpu;
    float * delta_gpu;