From 0f645836f193e75c4c3b718369e6fab15b5d19c5 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 11 Feb 2015 03:41:03 +0000 Subject: [PATCH] Detection is back, baby\! --- src/convolutional_layer.h | 13 ++++++++----- 1 files changed, 8 insertions(+), 5 deletions(-) diff --git a/src/convolutional_layer.h b/src/convolutional_layer.h index c69686f..72f3f72 100644 --- a/src/convolutional_layer.h +++ b/src/convolutional_layer.h @@ -45,28 +45,31 @@ void forward_convolutional_layer_gpu(convolutional_layer layer, float * in); void backward_convolutional_layer_gpu(convolutional_layer layer, float * in, float * delta_gpu); void update_convolutional_layer_gpu(convolutional_layer layer); + void push_convolutional_layer(convolutional_layer layer); void pull_convolutional_layer(convolutional_layer layer); -void learn_bias_convolutional_layer_ongpu(convolutional_layer layer); -void bias_output_gpu(const convolutional_layer layer); + +void bias_output_gpu(float *output, float *biases, int batch, int n, int size); +void backward_bias_gpu(float *bias_updates, float *delta, int batch, int n, int size); #endif convolutional_layer *make_convolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, int pad, ACTIVATION activation, float learning_rate, float momentum, float decay); -void resize_convolutional_layer(convolutional_layer *layer, int h, int w, int c); +void resize_convolutional_layer(convolutional_layer *layer, int h, int w); void forward_convolutional_layer(const convolutional_layer layer, float *in); void update_convolutional_layer(convolutional_layer layer); image *visualize_convolutional_layer(convolutional_layer layer, char *window, image *prev_filters); void backward_convolutional_layer(convolutional_layer layer, float *in, float *delta); -void bias_output(const convolutional_layer layer); +void bias_output(float *output, float *biases, int batch, int n, int size); +void backward_bias(float *bias_updates, float *delta, int batch, int n, int size); + image get_convolutional_image(convolutional_layer layer); image get_convolutional_delta(convolutional_layer layer); image get_convolutional_filter(convolutional_layer layer, int i); int convolutional_out_height(convolutional_layer layer); int convolutional_out_width(convolutional_layer layer); -void learn_bias_convolutional_layer(convolutional_layer layer); #endif -- Gitblit v1.10.0