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

gonna change im2col

Browse Source
This commit is contained in:
Joseph Redmon
2015-03-21 12:25:14 -07:00
parent dcb000b553
commit 4af116e996
16 changed files with 153 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/detection_layer.h
View File

@ -8,6 +8,7 @@ typedef struct {
int inputs;
int classes;
int coords;
int background;
int rescore;
float *output;
float *delta;
@ -17,7 +18,7 @@ typedef struct {
#endif
} detection_layer;
detection_layer *make_detection_layer(int batch, int inputs, int classes, int coords, int rescore);
detection_layer *make_detection_layer(int batch, int inputs, int classes, int coords, int rescore, int background);
void forward_detection_layer(const detection_layer layer, network_state state);
void backward_detection_layer(const detection_layer layer, network_state state);
int get_detection_layer_output_size(detection_layer layer);