route handles input images well....ish

This commit is contained in:
Joseph Redmon
2015-05-11 13:46:49 -07:00
parent dc0d7bb8a8
commit 516f019ba6
31 changed files with 1250 additions and 1819 deletions

View File

@ -5,37 +5,9 @@
#include "params.h"
#include "image.h"
#include "activations.h"
#include "layer.h"
typedef struct {
int batch;
int h,w,c;
int n;
int size;
int stride;
float *filters;
float *filter_updates;
float *biases;
float *bias_updates;
float *col_image;
float *delta;
float *output;
#ifdef GPU
float * filters_gpu;
float * filter_updates_gpu;
float * biases_gpu;
float * bias_updates_gpu;
float * col_image_gpu;
float * delta_gpu;
float * output_gpu;
#endif
ACTIVATION activation;
} deconvolutional_layer;
typedef layer deconvolutional_layer;
#ifdef GPU
void forward_deconvolutional_layer_gpu(deconvolutional_layer layer, network_state state);
@ -45,7 +17,7 @@ void push_deconvolutional_layer(deconvolutional_layer layer);
void pull_deconvolutional_layer(deconvolutional_layer layer);
#endif
deconvolutional_layer *make_deconvolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, ACTIVATION activation);
deconvolutional_layer make_deconvolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, ACTIVATION activation);
void resize_deconvolutional_layer(deconvolutional_layer *layer, int h, int w);
void forward_deconvolutional_layer(const deconvolutional_layer layer, network_state state);
void update_deconvolutional_layer(deconvolutional_layer layer, float learning_rate, float momentum, float decay);