mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Big changes to detection

Browse Source
This commit is contained in:
Joseph Redmon
2015-03-04 14:56:38 -08:00
parent 5f4a5f59b0
commit fb9e0fe336
17 changed files with 298 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/detection_layer.h
View File

@ -3,38 +3,26 @@
typedef struct {
int batch;
int h,w,c;
int n;
int size;
int stride;
float *filters;
float *filter_updates;
float *filter_momentum;
float *biases;
float *bias_updates;
float *bias_momentum;
float *col_image;
float *delta;
int inputs;
int classes;
int coords;
int rescore;
float *output;
float *delta;
#ifdef GPU
cl_mem filters_cl;
cl_mem filter_updates_cl;
cl_mem filter_momentum_cl;
cl_mem biases_cl;
cl_mem bias_updates_cl;
cl_mem bias_momentum_cl;
cl_mem col_image_cl;
cl_mem delta_cl;
cl_mem output_cl;
float * output_gpu;
float * delta_gpu;
#endif
} detection_layer;
ACTIVATION activation;
} convolutional_layer;
detection_layer *make_detection_layer(int batch, int inputs, int classes, int coords, int rescore);
void forward_detection_layer(const detection_layer layer, float *in, float *truth);
void backward_detection_layer(const detection_layer layer, float *in, float *delta);
int get_detection_layer_output_size(detection_layer layer);
#ifdef GPU
void forward_detection_layer_gpu(const detection_layer layer, float *in, float *truth);
void backward_detection_layer_gpu(detection_layer layer, float *in, float *delta);
#endif
#endif