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


@@ -2,8 +2,14 @@


#ifndef NETWORK_H


#define NETWORK_H




#include "image.h"


#include <stdint.h>


#include "layer.h"




#ifdef __cplusplus


extern "C" {


#endif




#include "image.h"


#include "data.h"


#include "tree.h"




@@ -15,7 +21,7 @@


    float *workspace;


    int n;


    int batch;


    int *seen;


   int *seen;


    float epoch;


    int subdivisions;


    float momentum;


@@ -51,6 +57,7 @@


    float exposure;


    float saturation;


    float hue;


   int small_object;




    int gpu_index;


    tree *hierarchy;


@@ -58,6 +65,11 @@


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




@@ -121,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