deconvolutional_layer.h
Go to the documentation of this file.
00001 #ifndef DECONVOLUTIONAL_LAYER_H
00002 #define DECONVOLUTIONAL_LAYER_H
00003 
00004 #include "cuda.h"
00005 #include "image.h"
00006 #include "activations.h"
00007 #include "layer.h"
00008 #include "network.h"
00009 
00010 typedef layer deconvolutional_layer;
00011 
00012 #ifdef GPU
00013 void forward_deconvolutional_layer_gpu(deconvolutional_layer layer, network_state state);
00014 void backward_deconvolutional_layer_gpu(deconvolutional_layer layer, network_state state);
00015 void update_deconvolutional_layer_gpu(deconvolutional_layer layer, float learning_rate, float momentum, float decay);
00016 void push_deconvolutional_layer(deconvolutional_layer layer);
00017 void pull_deconvolutional_layer(deconvolutional_layer layer);
00018 #endif
00019 
00020 deconvolutional_layer make_deconvolutional_layer(int batch, int h, int w, int c, int n, int size, int stride, ACTIVATION activation);
00021 void resize_deconvolutional_layer(deconvolutional_layer *layer, int h, int w);
00022 void forward_deconvolutional_layer(const deconvolutional_layer layer, network_state state);
00023 void update_deconvolutional_layer(deconvolutional_layer layer, float learning_rate, float momentum, float decay);
00024 void backward_deconvolutional_layer(deconvolutional_layer layer, network_state state);
00025 
00026 image get_deconvolutional_image(deconvolutional_layer layer);
00027 image get_deconvolutional_delta(deconvolutional_layer layer);
00028 image get_deconvolutional_filter(deconvolutional_layer layer, int i);
00029 
00030 int deconvolutional_out_height(deconvolutional_layer layer);
00031 int deconvolutional_out_width(deconvolutional_layer layer);
00032 
00033 #endif
00034 


rail_object_detector
Author(s):
autogenerated on Sat Jun 8 2019 20:26:29