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

multipl yolo morge

Browse Source
This commit is contained in:
Francesco Gatti
2019-02-06 22:24:01 +00:00
parent 0e97452460
commit 2c63bf05be
3 changed files with 115 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/Layer.h
+5 -4
View File
@@ -361,12 +361,13 @@ public:
dnnType *bias_h, *bias_d; //anchors
virtual dnnType* infer(dataDim_t &dim, dnnType* srcData);
int computeDetections(int w, int h, int netw, int neth, float thresh);
int computeDetections(Yolo::detection *dets, int &ndets, int w, int h, int netw, int neth, float thresh);
const int MAX_DETECTIONS = 256;
dnnType *predictions;
Yolo::detection *dets;
int detected;
static const int MAX_DETECTIONS = 256;
static Yolo::detection *allocateDetections(int nboxes, int classes);
static void mergeDetections(Yolo::detection *dets, int ndets, int classes);
};
/**