blame | history | patch | commit | commitdiff | ignore whitespace
AlexeyAB
2018-08-23 bb48b9c992e2f48bff2432a0387681cad0c98dec 
 src/detection_layer.h


@@ -1,30 +1,20 @@


#ifndef DETECTION_LAYER_H


#define DETECTION_LAYER_H




#include "params.h"


#include "layer.h"


#include "network.h"




typedef struct {


    int batch;


    int inputs;


    int classes;


    int coords;


    int rescore;


    float *output;


    float *delta;


    #ifdef GPU


    float * output_gpu;


    float * delta_gpu;


    #endif


} detection_layer;


typedef layer detection_layer;




detection_layer *make_detection_layer(int batch, int inputs, int classes, int coords, int rescore);


void forward_detection_layer(const detection_layer layer, network_state state);


void backward_detection_layer(const detection_layer layer, network_state state);


int get_detection_layer_output_size(detection_layer layer);


detection_layer make_detection_layer(int batch, int inputs, int n, int size, int classes, int coords, int rescore);


void forward_detection_layer(const detection_layer l, network_state state);


void backward_detection_layer(const detection_layer l, network_state state);


void get_detection_boxes(layer l, int w, int h, float thresh, float **probs, box *boxes, int only_objectness);


void get_detection_detections(layer l, int w, int h, float thresh, detection *dets);




#ifdef GPU


void forward_detection_layer_gpu(const detection_layer layer, network_state state);


void backward_detection_layer_gpu(detection_layer layer, network_state state);


void forward_detection_layer_gpu(const detection_layer l, network_state state);


void backward_detection_layer_gpu(detection_layer l, network_state state);


#endif




#endif