From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 16 Sep 2015 21:12:10 +0000 Subject: [PATCH] yolo working w/ regions --- src/convolutional_layer.h | 4 +++- 1 files changed, 3 insertions(+), 1 deletions(-) diff --git a/src/convolutional_layer.h b/src/convolutional_layer.h index 334759c..7452c3c 100644 --- a/src/convolutional_layer.h +++ b/src/convolutional_layer.h @@ -22,7 +22,7 @@ #endif convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, int pad, ACTIVATION activation); -void resize_convolutional_layer(convolutional_layer *layer, int h, int w); +void resize_convolutional_layer(convolutional_layer *layer, int w, int h); void forward_convolutional_layer(const convolutional_layer layer, network_state state); void update_convolutional_layer(convolutional_layer layer, int batch, float learning_rate, float momentum, float decay); image *visualize_convolutional_layer(convolutional_layer layer, char *window, image *prev_filters); @@ -38,6 +38,8 @@ int convolutional_out_height(convolutional_layer layer); int convolutional_out_width(convolutional_layer layer); +void rescale_filters(convolutional_layer l, float scale, float trans); +void rgbgr_filters(convolutional_layer l); #endif -- Gitblit v1.10.0