src/network.h
@@ -3,6 +3,7 @@ #define NETWORK_H #include "image.h" #include "detection_layer.h" #include "params.h" #include "data.h" @@ -23,6 +24,7 @@ int n; int batch; int seen; int subdivisions; float learning_rate; float momentum; float decay; @@ -80,6 +82,10 @@ void set_batch_network(network *net, int b); int get_network_input_size(network net); float get_network_cost(network net); detection_layer *get_network_detection_layer(network net); int get_network_nuisance(network net); int get_network_background(network net); #endif
