yolo_layer.h

#ifndef __YOLO_LAYER_H_
#define __YOLO_LAYER_H_
//#include <caffe/caffe.hpp>
#include <string>
#include <vector>

//using namespace caffe;

typedef struct{
    float x,y,w,h;
}box;

typedef struct{
    box bbox;
    int classes;
    float* prob;
    float* mask;
    float objectness;
    int sort_class;
}detection;

typedef struct layer{
    int batch;
    int total;
    int n,c,h,w;
    int out_n,out_c,out_h,out_w;
    int classes;
    int inputs,outputs;
    int *mask;
    float* biases;
    float* output;
    float* output_gpu;
}layer;

layer make_yolo_layer(int batch,int w,int h,int n,int total,int classes);

void free_yolo_layer(layer l);

void forward_yolo_layer_gpu(const void* input,layer l);

detection* get_detections(void** blobs,int img_w,int img_h,int* nboxes, int classNum);

void free_detections(detection *dets,int nboxes);




#endif