mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
route handles input images well....ish
This commit is contained in:
@ -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);
|
||||
|
Reference in New Issue
Block a user