blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-03-30 9fbd1936ebed466aad50c52989b92d5e3fedc099 
 src/network.h


@@ -2,19 +2,26 @@


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


@@ -28,22 +35,57 @@


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




    int gpu_index;


    tree *hierarchy;




    #ifdef GPU


    float **input_gpu;


    float **truth_gpu;


   float **input16_gpu;


   float **output16_gpu;


   size_t *max_input16_size;


   size_t *max_output16_size;


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


@@ -68,6 +110,7 @@


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


@@ -90,9 +133,15 @@


void set_batch_network(network *net, int b);


int get_network_input_size(network net);


float get_network_cost(network net);


detection *get_network_boxes(network *net, int w, int h, float thresh, float hier, int *map, int relative, int *num, int letter);


void free_detections(detection *dets, int n);




int get_network_nuisance(network net);


int get_network_background(network net);




#ifdef __cplusplus


}


#endif




#endif