blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-03-08 a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52 
 src/network.h


@@ -2,19 +2,27 @@


#ifndef NETWORK_H


#define NETWORK_H




#include "image.h"


#include "detection_layer.h"


#include <stdint.h>


#include "layer.h"




#ifdef __cplusplus


extern "C" {


#endif




#include "image.h"


#include "data.h"


#include "tree.h"




typedef enum {


    CONSTANT, STEP, EXP, POLY, STEPS, SIG


    CONSTANT, STEP, EXP, POLY, STEPS, SIG, RANDOM


} learning_rate_policy;




typedef struct {


typedef struct network{


    float *workspace;


    int n;


    int batch;


    int *seen;


   int *seen;


    float epoch;


    int subdivisions;


    float momentum;


    float decay;


@@ -27,22 +35,52 @@


    float gamma;


    float scale;


    float power;


    int time_steps;


    int step;


    int max_batches;


    float *scales;


    int   *steps;


    int num_steps;


    int burn_in;




    int adam;


    float B1;


    float B2;


    float eps;




    int inputs;


    int h, w, c;


    int max_crop;


    int min_crop;


    float angle;


    float aspect;


    float exposure;


    float saturation;


    float hue;




    int gpu_index;


    tree *hierarchy;




    #ifdef GPU


    float **input_gpu;


    float **truth_gpu;


   int wait_stream;


    #endif


} network;




typedef struct network_state {


    float *truth;


    float *input;


    float *delta;


    float *workspace;


    int train;


    int index;


    network net;


} network_state;




#ifdef GPU


float train_networks(network *nets, int n, data d, int interval);


void sync_nets(network *nets, int n, int interval);


float train_network_datum_gpu(network net, float *x, float *y);


float *network_predict_gpu(network net, float *input);


float * get_network_output_gpu_layer(network net, int i);


@@ -50,6 +88,7 @@


float *get_network_output_gpu(network net);


void forward_network_gpu(network net, network_state state);


void backward_network_gpu(network net, network_state state);


void update_network_gpu(network net);


#endif




float get_current_rate(network net);


@@ -66,11 +105,12 @@


float train_network(network net, data d);


float train_network_batch(network net, data d, int n);


float train_network_sgd(network net, data d, int n);


float train_network_datum(network net, float *x, float *y);




matrix network_predict_data(network net, data test);


float *network_predict(network net, float *input);


float network_accuracy(network net, data d);


float *network_accuracies(network net, data d);


float *network_accuracies(network net, data d, int n);


float network_accuracy_multi(network net, data d, int n);


void top_predictions(network net, int n, int *index);


float *get_network_output(network net);


@@ -88,10 +128,13 @@


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);




#ifdef __cplusplus


}


#endif




#endif