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

lots of stuff

Browse Source
This commit is contained in:
Joseph Redmon
2016-01-28 12:30:38 -08:00
parent 1578ec70d7
commit 913d355ec1
35 changed files with 913 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/network.c
View File

@ -8,6 +8,7 @@
#include "crop_layer.h"
#include "connected_layer.h"
#include "rnn_layer.h"
#include "local_layer.h"
#include "convolutional_layer.h"
#include "activation_layer.h"
@ -82,6 +83,8 @@ char *get_layer_string(LAYER_TYPE a)
return "deconvolutional";
case CONNECTED:
return "connected";
case RNN:
return "rnn";
case MAXPOOL:
return "maxpool";
case AVGPOOL:
@ -144,6 +147,8 @@ void forward_network(network net, network_state state)
forward_detection_layer(l, state);
} else if(l.type == CONNECTED){
forward_connected_layer(l, state);
} else if(l.type == RNN){
forward_rnn_layer(l, state);
} else if(l.type == CROP){
forward_crop_layer(l, state);
} else if(l.type == COST){
@ -178,6 +183,8 @@ void update_network(network net)
update_deconvolutional_layer(l, rate, net.momentum, net.decay);
} else if(l.type == CONNECTED){
update_connected_layer(l, update_batch, rate, net.momentum, net.decay);
} else if(l.type == RNN){
update_rnn_layer(l, update_batch, rate, net.momentum, net.decay);
} else if(l.type == LOCAL){
update_local_layer(l, update_batch, rate, net.momentum, net.decay);
}
@ -252,6 +259,8 @@ void backward_network(network net, network_state state)
if(i != 0) backward_softmax_layer(l, state);
} else if(l.type == CONNECTED){
backward_connected_layer(l, state);
} else if(l.type == RNN){
backward_rnn_layer(l, state);
} else if(l.type == LOCAL){
backward_local_layer(l, state);
} else if(l.type == COST){