mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

get regions

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-08 17:17:24 +02:00
parent 0a9957ba18
commit d7ce952465
4 changed files with 206 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/Layer.h
+13 -1
View File
@@ -272,6 +272,11 @@ public:
int stride;
};
struct box {
float x, y, w, h;
};
/**
Region layer
Mantain same dimension but change C*H*W distribution
@@ -279,16 +284,23 @@ public:
class Region : public Layer {
public:
Region(Network *net, int classes, int coords, int num, float thresh);
Region(Network *net, int classes, int coords, int num, float thresh, const char* fname_weights);
virtual ~Region();
virtual layerType_t getLayerType() { return LAYER_REGION; };
virtual value_type* infer(dataDim_t &dim, value_type* srcData);
value_type *bias_h, *bias_d;
int classes, coords, num;
float thresh;
int entry_index(int batch, int location, int entry);
box get_region_box(float *x, float *biases, int n, int index, int i, int j, int w, int h, int stride);
void get_region_boxes( float *input, int w, int h, int netw, int neth, float thresh,
float **probs, box *boxes, int only_objectness,
int *map, float tree_thresh, int relative);
void correct_region_boxes(box *boxes, int n, int w, int h, int netw, int neth, int relative);
void interpretData();
};