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

Added: weights_normalizion = relu or softmax for [shortcut] layer

Browse Source
This commit is contained in:
AlexeyAB
2020-01-09 20:54:09 +03:00
parent 0dbd0e1ef0
commit f1643cbf2e
8 changed files with 167 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/darknet.h
View File

@ -125,6 +125,10 @@ typedef enum {
NO_WEIGHTS, PER_FEATURE, PER_CHANNEL
} WEIGHTS_TYPE_T;
// parser.h
typedef enum {
NO_NORMALIZATION, RELU_NORMALIZATION, SOFTMAX_NORMALIZATION
} WEIGHTS_NORMALIZATION_T;
// image.h
typedef enum{
@ -336,6 +340,7 @@ struct layer {
float **layers_output;
float **layers_delta;
WEIGHTS_TYPE_T weights_type;
WEIGHTS_NORMALIZATION_T weights_normalizion;
int * map;
int * counts;
float ** sums;